PyTorch torch.matrix_power 関数
Pytorch torch リファレンスマニュアルtorch.matrix_powerPyTorch で行列のべき乗を計算するために使用される関数です。
関数定義
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)
その他の拡張