[ 返回参数类型不一致 ] torch.sparse.sum¶

torch.sparse.sum ¶

          torch.sparse.sum(input, dim=None, dtype=None)

         

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

         

输入参数不一致和返回类型不一致，具体如下：

PyTorch	PaddlePaddle	备注
input	x	输入的 Tensor，仅参数名不一致。
dim	axis	输入的第二个 Tensor，仅参数名不一致。
dtype	dtype	输出数据的类型。
-	keepdim	是否留减少的维度， PyTorch 无此参数， Paddle 保持默认即可。
返回值	返回值	当不指定 dim 时，PyTorch 返回 0D Tensor， Paddle 返回 Sparse Tensor。

           # PyTorch 写法
y = torch.sparse.sum(x)

# Paddle 写法
y = paddle.sparse.sum(x).values()