【Hackathon 4th No.30】为 Paddle 新增 paddle.sparse.sum 稀疏 API by zrr1999 · Pull Request #51406 · PaddlePaddle/Paddle

PR types

New features

PR changes

APIs

Description

Add paddle.sparse.sum
document pr: PaddlePaddle/docs#5846
rfc pr: PaddlePaddle/community#411

  • 基础要求
  • coo格式keep_dim参数支持
  • csr格式keep_dim参数支持(因为csr只支持2D或者3D格式,所以keep_dim为False时只会将Tensor变为2D)
  • dtype参数支持
  • 静态图

特殊说明
对于coo格式的稀疏张量,即使keep_dim为False,得到的结果张量的sparse_dim以及dense_dims.size()也会不小于1。
即当x.shape=[5,5],x.sparse_dim=1时,keep_dim=False
y=sum(x, axis=0)
z=sum(x, axis=1)
得到
y.shape=[1,5],y.sparse_dim=1
z.shape=[5,1],z.sparse_dim=1