netket.stats.online_statistics_batch

netket.stats.online_statistics_batch#

netket.stats.online_statistics_batch(data, combinator, old_estimator=None, *, decay=None, max_lag=64)[source]#

Accumulate K-channel local estimators into an OnlineStatsBatch.

Parameters:
  • data – shape (n_chains, chain_len, K)

  • combinator (Callable) – f: (K,) -> scalar, JAX-traceable

  • old_estimator (OnlineStatsBatch | None) – previous OnlineStatsBatch, or None to start fresh

  • decay (float | None) – EMA decay factor applied per update call (only used when creating a fresh accumulator; default None = no decay)

  • max_lag (int) – maximum ACF lag (only used when creating a fresh accumulator)

Return type:

OnlineStatsBatch

Returns:

Updated OnlineStatsBatch.