torch_geometric.transforms.PointPairFeatures

class PointPairFeatures(cat: bool = True)[source]

Bases: BaseTransform

Computes the rotation-invariant Point Pair Features (functional name: point_pair_features).

\[\left( \| \mathbf{d_{j,i}} \|, \angle(\mathbf{n}_i, \mathbf{d_{j,i}}), \angle(\mathbf{n}_j, \mathbf{d_{j,i}}), \angle(\mathbf{n}_i, \mathbf{n}_j) \right)\]

of linked nodes in its edge attributes, where \(\mathbf{d}_{j,i}\) denotes the difference vector between, and \(\mathbf{n}_i\) and \(\mathbf{n}_j\) denote the surface normals of node \(i\) and \(j\) respectively.

Parameters:

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