torch_geometric.transforms.BaseTransform — pytorch_geometric documentation (original) (raw)

class BaseTransform[source]

Bases: ABC

An abstract base class for writing transforms.

Transforms are a general way to modify and customizeData orHeteroData objects, either by implicitly passing them as an argument to a Dataset, or by applying them explicitly to individualData orHeteroData objects:

import torch_geometric.transforms as T from torch_geometric.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.