nip.graph_isomorphism.dataset_generation.shuffle_adjacencies#
- nip.graph_isomorphism.dataset_generation.shuffle_adjacencies(adjacencies: Int32[Tensor, '... node1 node2'], sizes: Int32[Tensor, '...']) Int32[Tensor, '... node1 node2'] [source]#
Shuffle the nodes in a batch of adjacency matrices of varying sizes.
Allows for arbitrary batch dimensions.
- Parameters:
adjacencies (Int32[Tensor, "... node1 node2"]) – The batch of adjacency matrices to shuffle.
sizes (Int32[Tensor, "..."]) – The number of nodes in each graph.
- Returns:
adjacencies (Int32[Tensor, “… node1 node2”]) – The batch of adjacency matrices with the nodes shuffled.