PyTorch torch.linalg.inv 函数

PyTorch torch.linalg.inv 函数 Pytorch torch 参考手册 torch.linalg.inv 是 PyTorch 线性代数模块中用于计算矩阵逆的函数。它是 torch.inverse 的推荐替代函数。函数定义 torch.linalg.inv(A) 参数: A (Tensor): 输入方阵，必须可逆。返回值: torch.Tensor: 返回矩阵的逆。使用示例实例..

更新于 2026/7/16

教程详情页默认 noindex。

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:
tensor([[1., 2.],
        [3., 4.]])
逆矩阵 A^-1:
tensor([[-2.0000,  1.0000],
        [ 1.5000, -0.5000]])
验证: A @ A^-1 =
tensor([[1., 0.],
        [0., 1.]])

Pytorch torch 参考手册