dgl.graphbolt.unique_and_compact

dgl.graphbolt.unique_and_compact(nodes: List[Tensor] | Dict[str, List[Tensor]])

压缩节点张量列表。

Parameters:

nodes (List[torch.Tensor] 或 Dict[str, List[torch.Tensor]]) – 用于压缩的节点列表。 unique_and_compact 将按类型执行 - 如果 nodes 是张量列表：所有张量将一起进行唯一化和压缩，通常用于同构图。 - 如果 nodes 是字典列表：键应为节点类型，值应为相应的节点，unique_and_compact 将按类型执行，通常用于异构图。

Returns:

输入中所有节点的唯一节点（按类型）。以及压缩后的节点列表，其中内部的ID被替换为压缩后的节点ID。“压缩节点列表”表示输入节点列表中的节点ID被替换为映射后的节点ID，其中每种类型的节点都被映射到一个从0到N的连续ID空间。

Return type:

元组[唯一节点, 压缩节点列表]