写文章
- @weixin_44526189
简介
该用户还未填写简介
擅长的技术栈
未填写擅长的技术栈
可提供的服务
暂无可提供的服务
PyTorch Geometric 实现简易 Graph Convolutional Networks (GCN)
GraphConv 层已经在 PyG 中预定义了，我们直接使用即可。这里我们使用两个 GraphConv 层，第一层的输入维度是每个节点的特征维度，输出维度为 16，第二个层的输入维度为 16，输出维度为节点类别数。每个层后面跟一个 ReLU 和 dropout 操作。device = torch.device('cuda' if torch.cuda.is_available() else 'c
到底了