pylibwholegraph.torch.wholegraph_ops.unweighted_sample_without_replacement

pylibwholegraph.torch.wholegraph_ops.unweighted_sample_without_replacement(

wm_csr_row_ptr_tensor: PyWholeMemoryTensor,

wm_csr_col_ptr_tensor: PyWholeMemoryTensor,

center_nodes_tensor: Tensor,

max_sample_count: int,

random_seed: int | None = None,

need_center_local_output: bool = False,

need_edge_output: bool = False,

)

Unweighted neighborhood sample in CSR WholeGraph