torch.lu_solve¶
- torch.lu_solve(b, LU_data, LU_pivots, *, out=None) Tensor ¶
使用
lu_factor()
中 A 的部分主元 LU 分解返回线性方程组 的 LU 解。此函数支持
float
、double
、cfloat
和cdouble
数据类型用于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) – 大小为 的 RHS 张量,其中 是零个或多个批次维度。
LU_data (Tensor) – 大小为 的 A 的主元 LU 分解,来自
lu_factor()
,其中 是零个或多个批次维度。LU_pivots (IntTensor) – 大小为 的 LU 分解的主元,来自
lu_factor()
,其中 是零个或多个批次维度。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)