PyTorch torch.linalg.inv 函数
Pytorch torch 参考手册torch.linalg.inv 是 PyTorch 线性代数模块中用于计算矩阵逆的函数。它是 torch.inverse 的推荐替代函数。
函数定义
torch.linalg.inv(A)
参数:
A(Tensor): 输入方阵，必须可逆。
返回值:
torch.Tensor: 返回矩阵的逆。
使用示例
实例
import torch
# 创建可逆矩阵
A = torch.tensor([[1.0, 2.0],
[3.0, 4.0]])
# 计算逆矩阵
A_inv = torch.linalg.inv(A)
print("矩阵 A:")
print(A)
print("n逆矩阵 A^-1:")
print(A_inv)
print("n验证: A @ A^-1 =")
print(A @ A_inv)
# 创建可逆矩阵
A = torch.tensor([[1.0, 2.0],
[3.0, 4.0]])
# 计算逆矩阵
A_inv = torch.linalg.inv(A)
print("矩阵 A:")
print(A)
print("n逆矩阵 A^-1:")
print(A_inv)
print("n验证: A @ A^-1 =")
print(A @ A_inv)
输出结果为：
矩阵 A:
tensor([[1., 2.],
[3., 4.]])
逆矩阵 A^-1:
tensor([[-2.0000, 1.0000],
[ 1.5000, -0.5000]])
验证: A @ A^-1 =
tensor([[1., 0.],
[0., 1.]])