pytorch计算张量的逆矩阵

要计算张量的逆矩阵，可以使用PyTorch中的torch.inverse()函数。下面是两个示例：

1. 计算单个矩阵张量的逆矩阵：

import torch

matrix = torch.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
inverse_matrix = torch.inverse(matrix)
print("Inverse matrix:\n", inverse_matrix)

2. 计算两个矩阵张量的逆矩阵：

import torch

A = torch.tensor([[1, 2, 3], [4, 5, 6]])
B = torch.tensor([[7, 8, 9], [10, 11, 12]])

# 将两个矩阵张量合并为一个3维张量
combined_tensor = torch.stack([A, B])

# 计算逆矩阵
inverse_tensor = torch.inverse(combined_tensor)

# 分别获取两个矩阵的逆矩阵
inverse_A = inverse_tensor[0]
inverse_B = inverse_tensor[1]

print("Inverse matrix of A:\n", inverse_A)
print("Inverse matrix of B:\n", inverse_B)