[ 仅参数名不一致 ]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 表示数据类型。