torch 逆矩阵
时间: 2023-07-19 19:10:21 浏览: 238
矩阵求导文档
在PyTorch中,可以使用`torch.inverse()`函数来计算一个矩阵的逆矩阵。具体使用方法如下:
```python
import torch
# 创建一个2x2的矩阵
mat = torch.tensor([[1., 2.], [3., 4.]])
# 计算矩阵的逆矩阵
inv_mat = torch.inverse(mat)
print(inv_mat)
```
输出结果为:
```
tensor([[-2.0000, 1.0000],
[ 1.5000, -0.5000]])
```
需要注意的是,`torch.inverse()`函数只能计算可逆矩阵的逆矩阵,如果矩阵不可逆,会抛出`RuntimeError`异常。
阅读全文