PyTorch torch.matrix_power 函数
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)