torch_geometric.transforms.BaseTransform

class BaseTransform[source]

Bases: ABC

用于编写转换的抽象基类。

变换是一种通用方式，用于修改和自定义 Data 或 HeteroData 对象，可以通过隐式将它们作为参数传递给 Dataset，或者通过显式地将它们应用于单个 Data 或 HeteroData 对象：

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.