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

torch.sum

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

paddle.sum

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

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

参数映射

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