dgl.distributed.load_partition_feats

dgl.distributed.load_partition_feats(part_config, part_id)[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.

Returns

  • Dict[str, Tensor] – Node features.

  • Dict[str, Tensor] – Edge features.