MiniBatchTransformer

class dgl.graphbolt.MiniBatchTransformer(datapipe, transformer=None)

基类: MapperIterDataPipe

一个用于操作 mini-batch 的 mini-batch transformer。

功能名称: transform。

参数:

• datapipe (DataPipe) – 数据管道。

• transformer – 应用于每个 minibatch 的函数，负责转换 minibatch。