tf_geometric:适用于TensorFlow 1.x和2.x的高效友好的图形神经网络库-源码

tf_geometric 适用于TensorFlow 1.x和2.x的高效友好的图形神经网络库。 受rusty1s / pytorch_geometric的启发，我们为TensorFlow构建了一个GNN库。 主页和文档 主页： ： 文档： : （） 论文： 高效友好 我们使用消息传递机制来实现图神经网络，它比基于密集矩阵的实现高效，并且比基于稀疏矩阵的实现更友好。 另外，我们为复杂的GNN操作提供了简单而优雅的API。 下面的示例构造一个图，并在其上应用多头图注意力网络（GAT）： # coding=utf-8 import numpy as np import tf_geometric as tfg import tensorflow as tf graph = tfg . Graph ( x = np . random . randn ( 5 , 20 ),