【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