[ 组合替代实现 ]torch.Tensor.cholesky_inverse

torch.Tensor.cholesky_inverse

torch.Tensor.cholesky_inverse(upper=False)

PaddlePaddle 目前无对应 API,可使用如下代码组合实现该 API。

转写示例

# Pytorch 写法
y = u.cholesky_inverse(upper)

# Python 写法
ut = paddle.transpose(u, perm=[1, 0])
if upper:
    y = paddle.linalg.inv(paddle.matmul(ut, u))
else:
    y = paddle.inverse(paddle.matmul(u, ut))