cugraph_pyg.sampler.sampler.SampleReader

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

Iterator that processes results from the cuGraph distributed sampler.

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

Constructs a new SampleReader.

Parameters:

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

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

Methods

__init__(base_reader)

Constructs a new SampleReader.