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

torch.unsqueeze

torch.unsqueeze(input,
                dim)

paddle.unsqueeze

paddle.unsqueeze(x,
                 axis,
                 name=None)

两者功能一致且参数用法一致,仅参数名不一致,具体如下:

参数映射

PyTorch PaddlePaddle 备注
input x 输入的 Tensor ,仅参数名不一致。
dim axis 表示要插入维度的位置,仅参数名不一致。