nip.graph_isomorphism.dataset_generation.generate_er_graphs#
- nip.graph_isomorphism.dataset_generation.generate_er_graphs(num_graphs: int, graph_size: int, edge_probability: float, device: device | str | int = 'cpu') Int32[Tensor, 'batch node1 node2'] [source]#
Generate a batch of Erdős-Rényi graphs.
- Parameters:
- Returns:
adjacency (Int32[Tensor, “batch node1 node2”]) – The batch of adjacency matrices for the generated graphs.