dgl.distributed.load_partition_feats

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

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

参数:
  • part_config (str) – 分区配置文件的路径。

  • part_id (int) – 分区 ID。

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

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

返回值:

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

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