DistEdgeDataLoader

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

基础类:DistDataLoader

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

它封装了一组边上的可迭代对象,生成消息流图(MFGs)列表,作为在分布式图上进行边分类、边回归和链接预测的小批量计算依赖。

所有参数的含义与单机版本的dgl.dataloading.DataLoader相同,除了第一个参数g,它必须是一个dgl.distributed.DistGraph

Parameters: