cugraph_pyg.sampler.sampler.HomogeneousSampleReader

class cugraph_pyg.sampler.sampler.HomogeneousSampleReader(base_reader: Iterator[Tuple[Dict[str, Tensor], int, int]])

Subclass of SampleReader that reads homogeneous output samples produced by the cuGraph distributed sampler.

__init__(base_reader: Iterator[Tuple[Dict[str, Tensor], int, int]])

Constructs a new HomogeneousSampleReader

Parameters:

base_reader: Iterator[Tuple[Dict[str, “torch.Tensor”], int, int]]

The iterator responsible for loading saved samples produced by the cuGraph distributed sampler.

Methods

__init__(base_reader)

Constructs a new HomogeneousSampleReader