MiniBatchTransformer

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

基础类：MapperIterDataPipe

一个用于操作小批量数据的小批量转换器。

功能名称: transform.

Parameters:

  • datapipe (DataPipe) – 数据管道。

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