![](https://img-blog.csdnimg.cn/direct/d54cf5f3130f47bc81b3b1bd58951a96.png?x-oss-process=image/resize,m_fixed,h_224,w_224)
ML-GNN
文章平均质量分 85
一些图神经网络的内容
__momo__
AI4Science & CV
展开
-
torch_geometric 使用 batch
torch.geometric的batch不支持CNN中广泛使用的batch维度。CNN中使用batch的数据具有。基于torch.geometric的data具有node和edge的特征,其中node的维度表示为。中起始点和终点的index都必须小于nodes的个数(因为默认从0开始)将数据转换成Batch格式,或者手动将node concate成。与GCNConv, SAGEConv不同的是,当使用。,edge_index concate成。,edge的维度表示为。原创 2023-12-11 21:30:23 · 1212 阅读 · 0 评论 -
如何理解图卷积网络GCN
可以调用中的GCNConv进行实现。if bias:else:# 1.消息传递# 1.对所有节点进行新的空间映射# 2.添加偏置= None:# 3.返回source、target信息,对应边的起点和终点# 4.获得度矩阵# 5.度矩阵归一化# 6.计算sqrt(deg(i)) * sqrt(deg(j))# 7.返回所有边的映射# 8.计算归一化后的节点特征return x_j# 2.消息聚合# 1.返回source、target信息,对应边的起点和终点。原创 2024-01-02 10:56:57 · 945 阅读 · 0 评论