[ 仅参数名不一致 ]torch.tensor_split

api 存在重载情况,分别如下:


torch.tensor_split

torch.tensor_split(input, indices, dim=0)

paddle.tensor_split

paddle.tensor_split(x, num_or_indices, axis=0, name=None)

其中 Paddle 相比 PyTorch 仅参数名不一致,具体如下:

参数映射

PyTorch PaddlePaddle 备注
input x 表示输入的 Tensor ,仅参数名不一致。
indices num_or_indices 表示分割的索引,仅参数名不一致。
dim axis 表示需要分割的维度,仅参数名不一致。

torch.tensor_split

torch.tensor_split(input, tensor_indices_or_sections, dim=0)

paddle.tensor_split

paddle.tensor_split(x, num_or_indices, axis=0, name=None)

paddle 当前无对应功能,功能缺失。


torch.tensor_split

torch.tensor_split(input, sections, dim=0)

paddle.tensor_split

paddle.tensor_split(x, num_or_indices, axis=0, name=None)

其中 Paddle 相比 PyTorch 仅参数名不一致,具体如下:

参数映射

PyTorch PaddlePaddle 备注
input x 表示输入的 Tensor ,仅参数名不一致。
sections num_or_indices 表示分割的数量,仅参数名不一致。
dim axis 表示需要分割的维度,仅参数名不一致。