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

torch.prod

torch.prod(input,
           dim=None,
           keepdim=False,
           dtype=None)

paddle.prod

paddle.prod(x,
            axis=None,
            keepdim=False,
            dtype=None,
            name=None)

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

参数映射

| PyTorch | PaddlePaddle | 备注 | | ————- | ———— | —————————————————— | | input | x | 表示输入的 Tensor ,仅参数名不一致。 | | dim | axis | 表示进行乘积运算的轴,仅参数名不一致。 | | keepdim | keepdim | 表示是否在输出 Tensor 中保留减小的维度。 | | dtype | dtype | 表示数据类型。 |