[ 仅参数名不一致 ]torch.tensor_split¶
torch.tensor_split¶
torch.tensor_split(input, indices_or_sections, dim=0)
paddle.tensor_split¶
paddle.tensor_split(x, num_or_indices, axis=0, name=None)
其中 Paddle 相比 PyTorch 仅参数名不一致，具体如下：
参数映射¶
| PyTorch | PaddlePaddle | 备注 |
|---|---|---|
| input | x | 表示输入的 Tensor ，仅参数名不一致。 |
| indices_or_sections | num_or_indices | 表示分割的数量或索引，仅参数名不一致。 |
| dim | axis | 表示需要分割的维度，仅参数名不一致。 |