今天看GAT论文的时候发现里面有一个奇怪的类,SpGAT。查看注释得知这个玩意实现了稀疏矩阵的乘法。这也不难想到,因为GAT网络里面结点和结点的连接一般是很稀疏的,如果直接矩阵乘法时间复杂度可能不小,所以我们可以直接定义稀疏矩阵相乘。

  

posted on 2020-11-01 15:35  何莫道  阅读(379)  评论(0编辑  收藏  举报