[ torch 参数更多 ]torch.linalg.lu_factor¶

torch.linalg.lu_factor ¶

          torch.linalg.lu_factor(A, *, bool pivot=True, out=None)

         

paddle.linalg.lu ¶

          paddle.linalg.lu(x, pivot=True, get_infos=False, name=None)

         

PyTorch 相比 Paddle 支持更多其他参数，具体如下：

参数映射¶

PyTorch	PaddlePaddle	备注
A	x	表示需要进行 LU 分解的输入 Tensor ，仅参数名不一致。
pivot	pivot	表示 LU 分解时是否进行旋转。
-	get_infos	表示是否返回分解状态信息， Paddle 保持默认即可。
out	-	表示输出的 Tensor 元组， Paddle 无此参数，需要转写。

转写示例¶

out：指定输出¶

           # PyTorch 写法
torch.linalg.lu_factor(A, out=(LU, pivots))

# Paddle 写法
y = paddle.linalg.lu(A)
paddle.assign(y[0], out[0]), paddle.assign(y[1], out[1])

          

[ torch 参数更多 ]torch.linalg.lu_factor¶

torch.linalg.lu_factor¶

paddle.linalg.lu¶

参数映射¶

转写示例¶

out：指定输出¶

torch.linalg.lu_factor ¶

paddle.linalg.lu ¶