MiniBatchTransformerΒΆ

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

Bases: torch.utils.data.datapipes.datapipe.IterDataPipe[torch.utils.data.datapipes.iter.callable.T_co]

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.