torch_geometric.transforms.LocalCartesian

class LocalCartesian(norm: bool = True, cat: bool = True, interval: Tuple[float, float] = (0.0, 1.0))

Bases: BaseTransform

Saves the relative Cartesian coordinates of linked nodes in its edge attributes (functional name: local_cartesian). Each coordinate gets neighborhood-normalized to a specified interval (\([0, 1]\) by default).

Parameters

• norm (bool, optional) – If set to False, the output will not be normalized. (default: True)

• cat (bool, optional) – If set to False, all existing edge attributes will be replaced. (default: True)

• interval ((float, float), optional) – A tuple specifying the lower and upper bound for normalization. (default: (0.0, 1.0))