pytorch学习记录：torch.inverse()

最新推荐文章于 2024-03-14 23:22:10 发布

guoqiangszu

最新推荐文章于 2024-03-14 23:22:10 发布

阅读量7k

点赞数 1

分类专栏： pytorch 文章标签： pytorch python

本文链接：https://blog.csdn.net/guoqiangszu/article/details/124337865

版权

pytorch 专栏收录该内容

6 篇文章 0 订阅

订阅专栏

pytorch doc介绍

torch.inverse()和torch.linalg.inv()一样。
在这里插入图片描述

>>> import torch
>>> A = torch.randn(4, 4)
>>> Ainv = torch.linalg.inv(A)
>>> torch.dist(A @ Ainv, torch.eye(4))
tensor(1.1921e-07)

>>> A = torch.randn(2, 3, 4, 4)  # Batch of matrices
>>> Ainv = torch.linalg.inv(A)
>>> torch.dist(A @ Ainv, torch.eye(4)))
tensor(1.9073e-06)

>>> A = torch.randn(4, 4, dtype=torch.complex128)  # Complex matrix
>>> Ainv = torch.linalg.inv(A)
>>> torch.dist(A @ Ainv, torch.eye(4))
tensor(7.5107e-16, dtype=torch.float64)