Compose

class dgl.transforms.Compose(transforms)[source]

基类: BaseTransform

创建由多个变换按顺序组成的变换。

参数:: 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]))