Samplers

cugraph_pyg.sampler.BaseSampler(sampler, data)	Adapt a distributed cuGraph sampler to PyG sampler interfaces.
cugraph_pyg.sampler.SampleIterator(data, ...)	Iterator that combines output graphs with their features to produce final output minibatches that can be fed into a GNN model.
cugraph_pyg.sampler.BaseDistributedSampler(...)	Base class for distributed graph sampling using cuGraph.
cugraph_pyg.sampler.DistributedNeighborSampler(...)	Sample graph neighborhoods across one or more GPUs.
cugraph_pyg.sampler.sampler.SampleReader(...)	Iterator that processes results from the cuGraph distributed sampler.
cugraph_pyg.sampler.sampler.HomogeneousSampleReader(...)	Subclass of SampleReader that reads homogeneous output samples produced by the cuGraph distributed sampler.
cugraph_pyg.sampler.sampler.HeterogeneousSampleReader(...)	Subclass of SampleReader that reads heterogeneous output samples produced by the cuGraph distributed sampler.
cugraph_pyg.sampler.io.BufferedSampleReader(...)	Iterate over buffered groups of sampling inputs.