[ 仅参数名不一致 ]torch.Tensor.squeeze

torch.Tensor.squeeze

torch.Tensor.squeeze(dim=None)

paddle.Tensor.squeeze

paddle.Tensor.squeeze(axis=None)

两者功能一致，仅参数名不一致，具体如下：

参数映射

PyTorch PaddlePaddle 备注
dim axis 输入一个或一列整数，代表要压缩的轴，仅参数名不一致。