dgl.graphbolt.unique_and_compactยถ

dgl.graphbolt.unique_and_compact(nodes: Union[List[torch.Tensor], Dict[str, List[torch.Tensor]]])[source]ยถ

Compact a list of nodes tensor.

Parameters

nodes (List[torch.Tensor] or Dict[str, List[torch.Tensor]]) โ€“ List of nodes for compacting. the unique_and_compact will be done per type - If nodes is a list of tensor: All the tensors will do unique and compact together, usually it is used for homogeneous graph. - If nodes is a list of dictionary: The keys should be node type and the values should be corresponding nodes, the unique and compact will be done per type, usually it is used for heterogeneous graph.

Returns

  • Tuple[unique_nodes, compacted_node_list]

  • The Unique nodes (per type) of all nodes in the input. And the compacted

  • nodes list, where IDs inside are replaced with compacted node IDs.

  • โ€Compacted node listโ€ indicates that the node IDs in the input node

  • list are replaced with mapped node IDs, where each type of node is

  • mapped to a contiguous space of IDs ranging from 0 to N.