dgl.distributed.load_partition_feats

dgl.distributed.load_partition_feats(part_config, part_id, load_nodes=True, load_edges=True)[source]

从分区加载节点/边特征数据。

Parameters:

  • part_config (str) – The path of the partition config file.

  • part_id (int) – The partition ID.

  • load_nodes (bool, optional) – 是否加载节点特征。如果 False，则返回 None。

  • load_edges (bool, 可选) – 是否加载边特征。如果 False，则返回 None。

Returns:

  • Dict[str, Tensor] or None – 节点特征。

  • Dict[str, Tensor] or None – 边特征。