pytorch张量求逆
时间: 2023-11-05 16:05:31 浏览: 193
python求矩阵的逆
要在PyTorch中求张量的逆,可以使用`torch.inverse()`函数。以下是一个示例:
```python
import torch
# 创建一个2x2的张量
A = torch.tensor([[1.0, 2.0], [3.0, 4.0]])
# 求逆
A_inv = torch.inverse(A)
print(A_inv)
```
输出:
```
tensor([[-2.0000, 1.0000],
[ 1.5000, -0.5000]])
```
阅读全文