Функция torch.matrix_power в PyTorch
Справочное руководство 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)
# Создать квадратную матрицу
A = torch.tensor([[1.0, 2.0], [3.0, 4.0]])
# Вычислить 3-ю степень матрицы
A_power = torch.matrix_power(A, 3)
print("Степень матрицы:")
print(A_power)
Другие расширения