PyTorch等同于Numpy的linalg.solve？

Question

我已经查看了它，并且有一些帖子，例如this one，建议使用torch.gesv，但是我似乎在PyTorch文档中找不到它。

Answer 1

Pytorch 现在有一个与 NumPy 非常相似的 linalg 部分。因此，您可以执行 torch.linalg.solve 并期待与 NumPy 相同的行为。

Answer 2

Pytorch提供功能torch.solve，其功能类似于numpy.linalg.solve。它将输出线性系统的解以及用于计算线性系统的LU分解。 更多信息和示例代码here。