torch_geometric.transforms.LinearTransformation

class LinearTransformation(matrix: Tensor)[source]

基础类:BaseTransform

使用离线计算的正方形变换矩阵转换节点位置 data.pos(功能名称:linear_transformation)。

Parameters:

matrix (Tensor) – 形状为 [D, D] 的张量,其中 D 对应于节点位置的维度。