Skip to main content

Module types

Module types 

Source
Expand description

Tensor 感知的节点和边类型

扩展现有的节点/边系统以原生支持 tensor 数据 用于图神经网络(GNN)和其他机器学习应用

Structs§

AdjacencyMatrix
邻接矩阵表示:用于图神经网络计算
BatchedNodeFeatures
批量节点特征:用于 mini-batch 处理
DegreeMatrix
度矩阵:对角矩阵,对角线元素为节点的度
GNMessage
图神经网络消息:在消息传递过程中使用
TensorEdge
Tensor 边:带有 tensor 数据的边包装器
TensorNode
Tensor 节点:带有 tensor 数据的节点包装器

Type Aliases§

EdgeFeatures
边特征张量:用于存储边的特征(如注意力权重)
HiddenState
图神经网络中的隐藏状态
NodeEmbedding
节点嵌入:用于图神经网络的节点表示
NodeFeatures
节点特征张量:用于存储节点的特征向量/矩阵