pylibwholegraph.torch.graph_ops.add_csr_self_loop

pylibwholegraph.torch.graph_ops.add_csr_self_loop(

csr_row_ptr_tensor: Tensor,

csr_col_ptr_tensor: Tensor,

)

Add self loop to sampled CSR graph NOTE: this function will not check if there is already self loop in the raw CSR graph. :param csr_row_ptr_tensor: CSR row pointer tensor :param csr_col_ptr_tensor: CSR column index tensor :return: CSR graph added self loop