[ 输入参数用法不一致 ]torch.Tensor.split¶

torch.Tensor.split ¶

          torch.Tensor.split(split_size_or_sections, dim=0)

         

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

         

PyTorch 的 split_size_or_sections 与 Paddle 的 num_or_sections 用法不同，具体如下：

PyTorch	PaddlePaddle	备注
dim	axis	表示需要分割的维度，仅参数名不一致。
split_size_or_sections	num_or_sections	torch：int 时表示块的大小， list 时表示块的大小; paddle： int 时表示块的个数， list 时表示块的大小。因此对于 int 时，两者用法不同，需要转写。

           # pytorch
x = torch.randn(8, 2)
y = x.split(4)

# paddle
x = paddle.randn([8, 2])
y = x.split(2)

          