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