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.