[ 仅参数名不同 ]torch.chunk¶

torch.chunk¶

torch.chunk(input,
            chunks,
            dim=0)

paddle.chunk¶

paddle.chunk(x,
             chunks,
             axis=0,
             name=None)

两者功能一致且参数用法一致，仅参数名不同，具体如下：

参数映射¶

| PyTorch | PaddlePaddle | 备注 | | ————- | ———— | —————————————————— | | input | x | 表示输入的 Tensor ，仅参数名不一致。 | | chunks | chunks | 表示将输入 Tensor 划分成多少个相同大小的子 Tensor。 | | dim | axis |表示需要分割的维度 ，仅参数名不一致。 |