分布式节点数据加载器

class dgl.dataloading.DistNodeDataLoader(g, nids, graph_sampler, device=None, **kwargs)[source]

Bases: DistDataLoader

用于分布式图存储的节点采样图数据加载器。

它在一组节点上封装了一个可迭代对象,生成消息流图(MFGs)列表,作为所述小批量在分布式图上的计算依赖。

All the arguments have the same meaning as the single-machine counterpart dgl.dataloading.DataLoader except the first argument g which must be a dgl.distributed.DistGraph.

Parameters: