16、循环图神经网络与图自编码器:原理、实现与应用

循环图神经网络与图自编码器:原理、实现与应用

1. 循环图神经网络基础

循环图神经网络(Recurrent Graph Neural Networks)在图分析任务中展现出强大的能力。在一些操作中,会对特征进行零填充,使其维度符合要求,代码如下:

zero_pad = torch.zeros((feature.shape[0], self._out_features - feature.shape[1]))
feature = torch.cat([feature, zero_pad], -1)

接着,会根据边的类型数量进行不同的处理:
- 若图只有一种边类型,会执行快速路径,更新节点特征。

if self._n_etypes == 1 and etypes is None:
    graph.ndata['h'] = self.linears[0](feature)
    graph.update_all(fn.copy_u('h', 'm'), fn.sum('m', 'a'))
    a = graph.ndata.pop('a')  # (N, D)
  • 若图有多种边类型,则会针对每种边类型进行操作。
else:
    graph.ndata['h'] = feature
    for i in range(self._n_etypes):
     
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符  | 博主筛选后可见
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值