Функция torch.matrix_power в PyTorch


Pytorch torch 参考手册Справочное руководство PyTorch torch

torch.matrix_powerявляется функцией в PyTorch, используемой для вычисления степеней матриц.

Определение функции

torch.matrix_power(input, n)

Пример использования

Пример

import torch

# Создать квадратную матрицу
A = torch.tensor([[1.0, 2.0], [3.0, 4.0]])

# Вычислить 3-ю степень матрицы
A_power = torch.matrix_power(A, 3)
print("Степень матрицы:")
print(A_power)

Pytorch torch 参考手册Справочное руководство PyTorch torch

Другие расширения