PyTorch torch.inverse 函数
Pytorch torch 参考手册torch.inverse 是 PyTorch 中用于计算矩阵逆的函数。
函数定义
torch.inverse(input, out)
使用示例
实例
import torch
A = torch.tensor([[4.0, 7.0], [2.0, 6.0]])
# 计算逆矩阵
A_inv = torch.inverse(A)
print("逆矩阵:")
print(A_inv)
# 验证: A @ A_inv ≈ I
print("A @ A_inv:")
print(torch.mm(A, A_inv))
A = torch.tensor([[4.0, 7.0], [2.0, 6.0]])
# 计算逆矩阵
A_inv = torch.inverse(A)
print("逆矩阵:")
print(A_inv)
# 验证: A @ A_inv ≈ I
print("A @ A_inv:")
print(torch.mm(A, A_inv))