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.