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

torch.linalg.cholesky

torch.linalg.cholesky(A,*,upper=False,out=None)

paddle.linalg.cholesky

paddle.linalg.cholesky(x,upper=False,name=None)

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

参数映射

PyTorch PaddlePaddle 备注
A x 表示输入参数为多维 Tensor,它的维度应该为 [, M, N],其中为零或更大的批次尺寸,并且最里面的两个维度上的矩阵都应为对称的正定矩阵,仅参数名不一致。
upper upper 表示是否返回上三角矩阵或下三角矩阵。
out - 表示输出的 Tensor ,Paddle 无此参数,需要转写。

转写示例

out: 输出的 Tensor

# PyTorch 写法
torch.linalg.cholesky(x, upper=False, out=output)


# Paddle 写法
paddle.assign(paddle.linalg.cholesky(x, upper=False),output)