torch.linalg.inv

torch.linalg.inv 是 PyTorch 中用于计算矩阵的逆的函数。它接受一个输入张量并返回逆矩阵的张量。如果输入矩阵不可逆，则会引发异常。

以下是使用 torch.linalg.inv 的示例代码：

import torch

# 定义一个 2x2 的矩阵
matrix = torch.tensor([[1., 2.], [3., 4.]])

# 计算矩阵的逆
inv_matrix = torch.linalg.inv(matrix)

print(inv_matrix)

输出:

tensor([[-2.0000,  1.0000],
        [ 1.5000, -0.5000]])