[ torch 参数更多 ]torch.linalg.solve_triangular

torch.linalg.solve_triangular

torch.linalg.solve_triangular(A, B, *, upper, left=True, unitriangular=False, out=None)

paddle.linalg.triangular_solve

paddle.linalg.triangular_solve(x, y, upper=True, transpose=False, unitriangular=False, name=None)

Pytorch 相比 Paddle 支持更多其他参数,具体如下:

参数映射

PyTorch PaddlePaddle 备注
A x 表示线性方程组左边的系数 Tensor ,仅参数名不一致。
B y 表示线性方程组右边的 Tensor ,仅参数名不一致。
upper upper 表示对系数 Tensor 取上三角还是下三角。
left transpose 表示是否对系数 Tensor 进行转置 ,仅参数名不一致。
unitriangular unitriangular 表示是否将系数 Tensor 对角线元素假设为 1 来求解方程。
out - 表示输出的 Tensor , Paddle 无此参数,需要进行转写。

转写示例

out:指定输出

# Pytorch 写法
torch.linalg.solve_triangular(A, B, upper, left, unitriangular, out=y)

# Paddle 写法
paddle.assign(paddle.linalg.triangular_solve(A, B, upper, left, unitriangular) , y)