nip.graph_isomorphism.dataset_generation._generate_isomorphic_graphs#
- nip.graph_isomorphism.dataset_generation._generate_isomorphic_graphs(config: GraphIsomorphicDatasetConfig, non_iso_adjacencies: Int32[Tensor, 'pair batch node1 node2'], non_iso_sizes: Int32[Tensor, 'batch']) tuple[Int32[Tensor, 'pair batch node1 node2'], Int32[Tensor, 'batch']] [source]#
Generate isomorphic graphs.
- Parameters:
config (GraphIsomorphicDatasetConfig) – The configuration for the dataset.
non_iso_adjacencies (Int32[Tensor, "pair batch node1 node2"], device="cpu") – The batch of adjacency matrices for the non-isomorphic graph pairs.
non_iso_sizes (Int32[Tensor, "batch"], device="cpu") – The number of nodes in each non-isomorphic graph pair.
- Returns:
adjacencies (Int32[Tensor, “pair batch node1 node2”], device=”cpu”) – The batch of adjacency matrices for the generated graphs.
sizes (Int32[Tensor, “batch”], device=”cpu”) – The number of nodes in each graph.