gammagl.transforms.BaseTransform

class BaseTransform

An abstract base class for writing transforms. Transforms are a general way to modify and customize Graph objects, either by implicitly passing them as an argument to a Dataset, or by applying them explicitly to individual Graph objects.

>>> import gammagl.transforms as T
>>> from gammagl.datasets import TUDataset
>>> transform = T.Compose([T.ToUndirected(), T.AddSelfLoops()])
>>> dataset = TUDataset(path, name='MUTAG', transform=transform)
>>> data = dataset[0]  # Implicitly transform data on every access.
>>> data = TUDataset(path, name='MUTAG')[0]
>>> data = transform(data)  # Explicitly transform data.