NodeShuffle

类 dgl.transforms.NodeShuffle[源码]

基类: BaseTransform

随机打乱节点顺序。

示例

>>> import dgl
>>> import torch
>>> from dgl import NodeShuffle

>>> transform = NodeShuffle()
>>> g = dgl.graph(([0, 1], [1, 2]))
>>> g.ndata['h1'] = torch.tensor([[1., 2.], [3., 4.], [5., 6.]])
>>> g.ndata['h2'] = torch.tensor([[7., 8.], [9., 10.], [11., 12.]])
>>> g = transform(g)
>>> print(g.ndata['h1'])
tensor([[5., 6.],
        [3., 4.],
        [1., 2.]])
>>> print(g.ndata['h2'])
tensor([[11., 12.],
        [ 9., 10.],
        [ 7.,  8.]])