[ 仅 paddle 参数更多 ]torch.dsplit¶

torch.dsplit¶

torch.dsplit(input,
        indices_or_sections)

paddle.split¶

paddle.split(x,
        num_or_sections,
        axis=0,
        name=None)

Paddle 相比 PyTorch 支持更多其他参数，具体如下：

参数映射¶

| PyTorch | PaddlePaddle | 备注 | | ————- | ———— | —————————————————— | | input | x | 输入多维 Tensor ，仅参数名不一致。 | | indices_or_sections | num_or_sections | 用于分割的 int 或 list 或 tuple ，仅参数名不一致。 | | - | axis | 表示需要分割的维度，PyTorch 无此参数，Paddle 需要设置为 2。 |