MiniBatchTransformerΒΆ
-
class
dgl.graphbolt.
MiniBatchTransformer
(datapipe, transformer)[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.