Metrics

Functions for computing physical observables and metrics from sampled states.

isax.metrics.magnetization_per_site(state: list[jaxtyping.Int[Array, 'samples block_nodes']]) -> list[jaxtyping.Float[Array, block_nodes]]

Compute magnetization per site.

\[m_j = \frac{1}{N} \sum_{i=1}^{N} s_{i,j}\]

Arguments:

• state: List of arrays, shape (samples, block_nodes) per block.

Returns:

List of arrays, shape (block_nodes,) per block.