PyTorch中等价于Numpy的linalg.solve函数是什么？

Question

3

我已经查过了，有一些帖子，例如这篇，建议使用torch.gesv，但我在PyTorch文档中找不到它。

- Rani

2个回答

4

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

- Rahul Vigneswaran

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- trialNerror · Accepted Answer

Pytorch提供了torch.solve函数，其行为类似于numpy.linalg.solve。它将输出线性系统的解以及用于计算解的LU分解。

更多信息和示例代码在此处。