data.batch[i]的值表示:在该批次中,第i个节点所属的子图的索引号。例如,对于该批次中的第1个Data数据(记为D1),如果与D1相关的节点有n个(n=D1.x.size(0)),那么data.batch[0:n]的值都是0. 对于该批次中的第2个Data数据(记为D2),如果与D2相关的节点有m个(m=D2.x.size(0)),那么data.batch[...
name='AIDS')#下载数据data=dataset#TUDataset数据集data1=dataset[0]#第一个图数据data>>AIDS(2000)#AIDS这个数据集有2000个图data1>>Data(edge_index=[2,106],x=[47,38],edge_attr=[106,3],y=[1])#AIDS数据集中其中一个图的构成;#edge_index=[2,106]表示连通性2表示两个节点相连,106表示边的数...
retain_variables(bool) –True,计算梯度时所需要的buffer在计算完梯度后不会被释放。如果想对一个子图多次求微分的话,需要设置为True。 2. torch.autograd.backward()一般程序示例 求 中y对x的梯度: x.grad.zero_() #梯度初始化,防止梯度累计 x = np.arange(4.0, requires_grad = True) #x需要求梯度 y...
2.adjs.EdgeIndex.edge_index:是根据n_id的节点序号重新生成的子图的edge_index; 3.adjs.EdgeIndex.e_id:是在原图中的边id。举例:edge_index=tensor([[1, 2, 3, 3],[0, 0, 0, 1]])的第一组边 (1,0) pair以及n_id=tensor([2, 3, 1, 0]),即第一组边在原图中对应的节点pair为为(3,2...