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

torch.Tensor.squeeze

torch.Tensor.squeeze(dim=None)

paddle.Tensor.squeeze_

paddle.Tensor.squeeze(axis=None)

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

参数映射

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