From 51a244d7ef023f62a313af87ac7a2357f96b2cea Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=BC=A0=E6=8C=AF=E5=9D=A4?= <113668869+Zhang-Zhenkun@users.noreply.github.com> Date: Sat, 29 Jun 2024 19:41:40 +0800 Subject: [PATCH] Update torch.tensor_split.md MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit indices_or_sections必须分开使用,indices和sections均可作为位置参数或关键字参数使用,num_or_indices不可分开使用。 --- .../api_difference/ops/torch.tensor_split.md | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/docs/guides/model_convert/convert_from_pytorch/api_difference/ops/torch.tensor_split.md b/docs/guides/model_convert/convert_from_pytorch/api_difference/ops/torch.tensor_split.md index a6f9f6bb1bd..124a81d01ba 100644 --- a/docs/guides/model_convert/convert_from_pytorch/api_difference/ops/torch.tensor_split.md +++ b/docs/guides/model_convert/convert_from_pytorch/api_difference/ops/torch.tensor_split.md @@ -1,4 +1,4 @@ -## [ 仅参数名不一致 ]torch.tensor_split +## [ 参数用法不一致 ]torch.tensor_split ### [torch.tensor_split](https://pytorch.org/docs/stable/generated/torch.tensor_split.html) ```python @@ -11,12 +11,12 @@ torch.tensor_split(input, indices_or_sections, dim=0) paddle.tensor_split(x, num_or_indices, axis=0, name=None) ``` -其中 Paddle 相比 PyTorch 仅参数名不一致,具体如下: +其中 Paddle 相比 PyTorch 参数用法不一致,具体如下: ### 参数映射 | PyTorch | PaddlePaddle | 备注 | | ------------- | ------------ | ------------------------------------------------------ | | input | x | 表示输入的 Tensor ,仅参数名不一致。 | -| indices_or_sections | num_or_indices | 表示分割的数量或索引,仅参数名不一致。 | +| indices_or_sections | num_or_indices | indices_or_sections必须分开使用,indices和sections均可作为位置参数或关键字参数使用,num_or_indices不可分开使用。 | | dim | axis | 表示需要分割的维度,仅参数名不一致。 |