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 | 表示需要分割的维度,仅参数名不一致。 |