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
。