设置边缘延迟功能

class dgl.dataloading.base.set_edge_lazy_features(g, feature_names)[source]

基础类：

将懒加载特征分配给输入图的edata以进行预取优化。

When used in a Sampler, lazy features mark which data should be fetched before computation in model. See guide-minibatch-prefetching for a detailed explanation.

如果图是同质的，这相当于：

g.edata.update({k: LazyFeature(k, g.edata[dgl.EID]) for k in feature_names})

如果图是异构的，这相当于：

for type_, names in feature_names.items():
    g.edges[type_].data.update(
        {k: LazyFeature(k, g.edges[type_].data[dgl.EID]) for k in names})

Parameters:

g (DGLGraph) – The graph.
feature_names (list[str] or dict[etype, list[str]]) – 要预取的特征名称。etype 键是一个字符串或三元组。

另请参阅

dgl.LazyFeature