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

torch.unsqueeze

torch.unsqueeze(input,
                dim)

paddle.unsqueeze

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

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

参数映射

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