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) – 从
lu_factor()
获得的 A 的部分主元 LU 分解,大小为 ,其中 表示零个或多个批处理维度。LU_pivots (IntTensor) – 从
lu_factor()
获得的 LU 分解的主元,大小为 ,其中 表示零个或多个批处理维度。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)