[ 仅参数名不一致 ] torch.Tensor.chunk¶
torch.Tensor.chunk¶
torch.Tensor.chunk(chunks, dim=0)
参数映射¶
| PyTorch | PaddlePaddle | 备注 | |————| ———— | —————————————————— | | chunks | chunks | 表示将输入 Tensor 划分成多少个相同大小的子 Tensor。 | | dim | axis | 表示需要分割的维度,仅参数名不一致。 |