nip.graph_isomorphism.dataset_generation._generate_isomorphic_graphs

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.