PyTorch 相当于 Numpy 的 linalg.solve 是什么？

【问题标题】：What is the PyTorch equivalent to Numpy's linalg.solve?PyTorch 相当于 Numpy 的 linalg.solve 是什么？
【发布时间】：2021-01-03 15:45:46
【问题描述】：

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

【问题讨论】：

【解决方案1】：

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

【讨论】：

【解决方案2】：

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

【讨论】：