[ 返回参数类型不一致 ]torch.linalg.inv_ex

torch.linalg.inv_ex

torch.linalg.inv_ex(A, *, check_errors=False, out=None)

paddle.linalg.inv

paddle.linalg.inv(x, name=None)

两者功能一致但返回参数类型不同，具体如下：

参数映射

PyTorch PaddlePaddle 备注
A x 输入 Tensor，仅参数名不一致。
check_errors - 是否检查错误，Paddle 无此参数，一般对网络训练结果影响不大，可直接删除。
out - 表示输出的 Tensor ， Paddle 无此参数，需要转写。
返回值 返回值 Pytorch 返回两个 out 与 info，Paddle 仅返回一个 Tensor：out，需转写。

转写示例

返回值

# PyTorch 写法
torch.linalg.inv_ex(x)

# Paddle 写法
(paddle.linalg.inv(x), paddle.zeros(x.shape[:-2], dtype='int32'))

out 参数：输出的 Tensor

# PyTorch 写法
torch.linalg.inv_ex(x, out=y)

# Paddle 写法
out1 = paddle.linalg.inv(x)
out2 = paddle.zeros(x.shape[:-2], dtype='int32')
paddle.assign(out1, output=y[0]), paddle.assign(out2, output=y[1])