-
Lingfan Yu authored
* implement pytorch spmm with gather and scatter add * fix * replace torch take with index_select * comments * comment about pytorch __getitem__ operator pitfall * typo
3cc32a97
* implement pytorch spmm with gather and scatter add * fix * replace torch take with index_select * comments * comment about pytorch __getitem__ operator pitfall * typo