快捷方式

torch.lu_solve

torch.lu_solve(b, LU_data, LU_pivots, *, out=None) Tensor

使用从 lu_factor() 获得的 A 的部分主元 LU 分解,返回线性方程组 Ax=bAx = b 的 LU 解。

该函数支持 float, double, cfloatcdouble 数据类型的 input

警告

torch.lu_solve() 已被弃用,推荐使用 torch.linalg.lu_solve()torch.lu_solve() 将在未来的 PyTorch 版本中移除。X = torch.lu_solve(B, LU, pivots) 应替换为

X = linalg.lu_solve(LU, pivots, B)
参数
  • b (Tensor) – 大小为 (,m,k)(*, m, k) 的 RHS 张量,其中 * 表示零个或多个批处理维度。

  • LU_data (Tensor) – 从 lu_factor() 获得的 A 的部分主元 LU 分解,大小为 (,m,m)(*, m, m),其中 * 表示零个或多个批处理维度。

  • LU_pivots (IntTensor) – 从 lu_factor() 获得的 LU 分解的主元,大小为 (,m)(*, m),其中 * 表示零个或多个批处理维度。LU_pivots 的批处理维度必须与 LU_data 的批处理维度相等。

关键字参数

out (Tensor, 可选) – 输出张量。

示例

>>> A = torch.randn(2, 3, 3)
>>> b = torch.randn(2, 3, 1)
>>> LU, pivots = torch.linalg.lu_factor(A)
>>> x = torch.lu_solve(b, LU, pivots)
>>> torch.dist(A @ x, b)
tensor(1.00000e-07 *
       2.8312)

文档

查阅 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发者的深入教程

查看教程

资源

查找开发资源并获得问题解答

查看资源