gammagl.loader.NeighborSampler

class NeighborSampler(edge_index: CommonTensor | SparseGraph, sample_lists: List[int], node_idx: CommonTensor | None = None, num_nodes: int | None = None, return_e_id: bool = True, transform: Callable | None = None, **kwargs)[source]
sample(batch)[source]