torch_geometric.transforms.Cartesian

class Cartesian(norm: bool = True, max_value: Optional[float] = None, cat: bool = True)[source]

Bases: BaseTransform

Saves the relative Cartesian coordinates of linked nodes in its edge attributes (functional name: cartesian).

Parameters

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

  • max_value (float, optional) – If set and norm=True, normalization will be performed based on this value instead of the maximum value found in the data. (default: None)

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