neuralqx.nn.projectors.utils module

symmetry_to_index_perm(symmetry_obj, graph)

Convert a user symmetry object into an edge-index permutation using graph.

Return type:

Sequence[int]

Requirements on graph:
  • exposes edges (iterable of canonical edges)

  • exposes edge_to_index(edge) -> int

Requirements on symmetry_obj:
  • defines map_edge(edge) -> edge'