DistEdgeDataLoader
- class dgl.dataloading.DistEdgeDataLoader(g, eids, graph_sampler, device=None, **kwargs)[source]
基础类:
DistDataLoader用于分布式图存储的边缘采样图数据加载器。
它封装了一组边上的可迭代对象,生成消息流图(MFGs)列表,作为在分布式图上进行边分类、边回归和链接预测的小批量计算依赖。
所有参数的含义与单机版本的
dgl.dataloading.DataLoader相同,除了第一个参数g,它必须是一个dgl.distributed.DistGraph。- Parameters:
g (DistGraph) – 分布式图。
eids – 参见
dgl.dataloading.DataLoader。graph_sampler – 参见
dgl.dataloading.DataLoader。设备 – 参见
dgl.dataloading.DataLoader.kwargs – 参见
dgl.dataloading.DataLoader。