torch_geometric.transforms.LocalCartesian

class LocalCartesian(norm: bool = True, cat: bool = True)

Bases: BaseTransform

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

Parameters

• norm (bool, optional) – If set to False, the output will not be normalized to the interval \({[0, 1]}^D\). (default: True)

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