netket.experimental.nn.rnn.get_snake_inv_reorder_idx

netket.experimental.nn.rnn.get_snake_inv_reorder_idx(graph)

A helper function to generate the inverse reorder indices in the snake order for a 2D graph.

See netket.experimental.models.RNN for details about the reordering indices.

Return type:

HashableArray

Parameters:

graph (AbstractGraph)