MiniBatchTransformer

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

Bases: MapperIterDataPipe

A mini-batch transformer used to manipulate mini-batch.

Functional name: transform.

Parameters:

  • datapipe (DataPipe) – The datapipe.

  • transformer – The function applied to each minibatch which is responsible for transforming the minibatch.