• Lingfan Yu's avatar
    Reduce API (#15) · 68fb5f7e
    Lingfan Yu authored
    * add reduce_msg related api to dgl graph
    
    * add reduce_sum, switch backend from numpy to pytorch
    
    * update gat gcn to use reduce msg api
    
    * remove reduce_sum
    
    * add built-in reduce functions
    68fb5f7e
gat.py 6.66 KB