组合

class dgl.transforms.Compose(transforms)

Bases: BaseTransform

创建一个由多个按顺序排列的转换组成的转换。

Parameters:

transforms (list of Callable) – 一个按顺序应用的转换对象列表。转换对象应继承 BaseTransform 并实现 __call__()。

示例

>>> import dgl
>>> from dgl import transforms as T

>>> g = dgl.graph(([0, 0], [1, 1]))
>>> transform = T.Compose([T.ToSimple(), T.AddReverse()])
>>> new_g = transform(g)
>>> print(new_g.edges())
(tensor([0, 1]), tensor([1, 0]))