torch_geometric.transforms.PointPairFeatures

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

Bases: BaseTransform

计算旋转不变的点对特征 (功能名称: 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)\]

在其边缘属性中的链接节点,其中 \(\mathbf{d}_{j,i}\) 表示之间的差异向量,而 \(\mathbf{n}_i\)\(\mathbf{n}_j\) 分别表示节点 \(i\)\(j\) 的表面法线。

Parameters:

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