\u200E

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

torch.squeeze ¶

          torch.squeeze(input,
              dim=None)

paddle.squeeze ¶

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

         

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

参数映射¶

PyTorch	PaddlePaddle	备注
input	x	表示输入的 Tensor ，仅参数名不一致。
dim	axis	表示要压缩的轴，仅参数名不一致。