netket_foundation.stats.replica_statistics

netket_foundation.stats.replica_statistics(data)

Per-replica netket.stats.statistics(), vmapped over the replica axis.

Parameters:

data (Array) – array of shape (n_replicas, n_chains_per_replica, n_samples).

Return type:

ReplicaStats

Returns:

A ReplicaStats with one scalar Stats worth of fields per replica.