我已经查过了,有一些帖子,例如这篇,建议使用torch.gesv,但我在PyTorch文档中找不到它。
torch.gesv
Pytorch提供了torch.solve函数,其行为类似于numpy.linalg.solve。它将输出线性系统的解以及用于计算解的LU分解。
torch.solve
numpy.linalg.solve
Pytorch现在有一个与NumPy非常相似的linalg部分。 因此,您可以执行torch.linalg.solve并期望与NumPy相同的行为。