netket.nn.states_to_numbers#

netket.nn.states_to_numbers(hilbert, σ)[source]#

Converts the configuration σ to a 64-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.

Note

Requires jax >= 0.3.17 and will raise an exception on older versions.

Parameters

  • hilbert (DiscreteHilbert) – The Hilbert space

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

Return type

Union[ndarray, DeviceArray, Tracer]

Returns

a single integer or a batch of integer indices.