dgl_sparse/include/sparse/softmax.h · 5ffd2a022c3bc2b37a47d460d5f32d034f61d81e · OpenDAS / dgl

[Sparse] Support column-wise softmax (#5377) · 5ffd2a02

czkkkkkk authored Feb 24, 2023



* [Sparse] Support column-wise softmax

* Update python/dgl/sparse/softmax.py
Co-authored-by: Mufei Li <mufeili1996@gmail.com>

---------
Co-authored-by: Mufei Li <mufeili1996@gmail.com>

5ffd2a02

softmax.h 818 Bytes

Replace softmax.h