From 9af79434777188d1d0d37a4a59a9c9b75ab1d493 Mon Sep 17 00:00:00 2001 From: co63oc Date: Tue, 29 Aug 2023 11:30:17 +0800 Subject: [PATCH] =?UTF-8?q?=E6=98=A0=E5=B0=84=E6=96=87=E6=A1=A3=20No.358/3?= =?UTF-8?q?61=20(#6139)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit * Add api diff * Fix --- .../linalg/torch.linalg.lu_factor.md | 37 ++++++++++++++++++ .../linalg/torch.linalg.lu_factor_ex.md | 38 +++++++++++++++++++ 2 files changed, 75 insertions(+) create mode 100644 docs/guides/model_convert/convert_from_pytorch/api_difference/linalg/torch.linalg.lu_factor.md create mode 100644 docs/guides/model_convert/convert_from_pytorch/api_difference/linalg/torch.linalg.lu_factor_ex.md diff --git a/docs/guides/model_convert/convert_from_pytorch/api_difference/linalg/torch.linalg.lu_factor.md b/docs/guides/model_convert/convert_from_pytorch/api_difference/linalg/torch.linalg.lu_factor.md new file mode 100644 index 00000000000..3deca13af55 --- /dev/null +++ b/docs/guides/model_convert/convert_from_pytorch/api_difference/linalg/torch.linalg.lu_factor.md @@ -0,0 +1,37 @@ +## [ torch 参数更多 ]torch.linalg.lu_factor + +### [torch.linalg.lu_factor](https://pytorch.org/docs/stable/generated/torch.linalg.lu_factor.html#torch.linalg.lu_factor) + +```python +torch.linalg.lu_factor(A, *, bool pivot=True, out=None) +``` + +### [paddle.linalg.lu](https://www.paddlepaddle.org.cn/documentation/docs/zh/develop/api/paddle/linalg/lu_cn.html) + +```python +paddle.linalg.lu(x, pivot=True, get_infos=False, name=None) +``` + +Pytorch 相比 Paddle 支持更多其他参数,具体如下: + +### 参数映射 + +| PyTorch | PaddlePaddle | 备注 | +| ------- | ------------ | ----------------------------------------------------- | +| A | x | 表示需要进行 LU 分解的输入 Tensor ,仅参数名不一致。 | +| pivot | pivot | 表示 LU 分解时是否进行旋转。 | +| - | get_infos | 表示是否返回分解状态信息 , Paddle 保持默认即可。 | +| out | - | 表示输出的 Tensor 元组 , Paddle 无此参数,需要转写。 | + +### 转写示例 + +#### out:指定输出 + +```python +# Pytorch 写法 +torch.linalg.lu_factor(A, out=(LU, pivots)) + +# Paddle 写法 +y = paddle.linalg.lu(A) +paddle.assign(y[0], out[0]), paddle.assign(y[1], out[1]) +``` diff --git a/docs/guides/model_convert/convert_from_pytorch/api_difference/linalg/torch.linalg.lu_factor_ex.md b/docs/guides/model_convert/convert_from_pytorch/api_difference/linalg/torch.linalg.lu_factor_ex.md new file mode 100644 index 00000000000..29d46ae3687 --- /dev/null +++ b/docs/guides/model_convert/convert_from_pytorch/api_difference/linalg/torch.linalg.lu_factor_ex.md @@ -0,0 +1,38 @@ +## [ torch 参数更多 ]torch.linalg.lu_factor_ex + +### [torch.linalg.lu_factor_ex](https://pytorch.org/docs/stable/generated/torch.linalg.lu_factor_ex.html?highlight=lu_factor_ex#torch.linalg.lu_factor_ex) + +```python +torch.linalg.lu_factor_ex(A, *, pivot=True, check_errors=False, out=None) +``` + +### [paddle.linalg.lu](https://www.paddlepaddle.org.cn/documentation/docs/zh/develop/api/paddle/linalg/lu_cn.html) + +```python +paddle.linalg.lu(x, pivot=True, get_infos=True, name=None) +``` + +Pytorch 相比 Paddle 支持更多其他参数,具体如下: + +### 参数映射 + +| PyTorch | PaddlePaddle | 备注 | +| ------------ | ------------ | --------------------------------------------------------------------------------- | +| A | x | 表示需要进行 LU 分解的输入 Tensor ,仅参数名不一致。 | +| pivot | pivot | 表示 LU 分解时是否进行旋转。 | +| - | get_infos | 表示是否返回分解状态信息 ,PyTorch 返回 infos 信息,Paddle 需要设置为 True。 | +| check_errors | - | 检查 infos 的内容,如果为非 0 抛出错误, Paddle 无此参数,一般对网络训练结果影响不大,可直接删除。 | +| out | - | 表示输出的 Tensor 元组 ,Paddle 无此参数,需要转写。 | + +### 转写示例 + +#### out:指定输出 + +```python +# Pytorch 写法 +torch.linalg.lu_factor_ex(A, out=(LU, pivots, info)) + +# Paddle 写法 +y = paddle.linalg.lu(A, get_infos=True) +paddle.assign(y[0], out[0]), paddle.assign(y[1], out[1]), paddle.assign(y[2], out[2]) +```