[ 组合替代实现 ]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))