[ 仅 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)