[ 仅参数名不一致 ]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 | 表示要压缩的轴，仅参数名不一致。 |