dgl.distributed.load_partition_feats

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

Load node/edge feature data from a partition.

Parameters

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

  • part_id (int) – The partition ID.

  • load_nodes (bool, optional) – Whether to load node features. If False, None is returned.

  • load_edges (bool, optional) – Whether to load edge features. If False, None is returned.

Returns

  • Dict[str, Tensor] or None – Node features.

  • Dict[str, Tensor] or None – Edge features.