[ 仅参数名不一致 ]torch.vsplit

torch.vsplit

torch.vsplit(input,
        indices_or_sections)

paddle.vsplit

paddle.vsplit(x,
        num_or_indices,
        name=None)

其中 Paddle 相比 Pytorch 仅参数名不一致,具体如下:

参数映射

PyTorch PaddlePaddle 备注
input x 输入多维 Tensor ,仅参数名不一致。
indices_or_sections num_or_indices 表示分割的数量或索引,仅参数名不一致。