tests/python/pytorch/sparse/test_softmax.py · 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

test_softmax.py 1.09 KB

Replace test_softmax.py