[ 仅 API 调用方式不一致 ]torch.split

torch.split

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

paddle.compat.split

paddle.compat.split(tensor, split_size_or_sections, dim=0)

两者功能一致,但调用方式不一致,具体如下:

转写示例

# PyTorch 写法
result = torch.split(a, 2)

# Paddle 写法
result = paddle.compat.split(a, 2)