netket.nn.states_to_numbers

netket.nn.states_to_numbers(hilbert, σ)

Converts the configuration σ to a 32-bit integer denoting its index in the full Hilbert space.

This function calls hilbert.states_to_numbers as a JAX pure callback and can thus be used within jax.jit.

Parameters:

• hilbert (DiscreteHilbert) – The Hilbert space

• σ (Union[ndarray, Array]) – A single or a batch of configurations

Return type:

Union[ndarray, Array]

Returns:

a single integer or a batch of integer indices.