prepare_gaussian_bin

anri.fwd.prepare_gaussian_bin(cov)

Precompute per-axis scaling factors for sample_gaussian_bins().

Parameters:

cov (Array) – [n, n] covariance matrix of the multivariate Gaussian.

Returns:

erf_scale (jax.Array) – [n] per-axis scaling factor \(s_i = \frac{1}{\sigma_i \sqrt{2}}\), where \(\sigma_i = \sqrt{\Sigma_{ii}}\) is the marginal standard deviation along axis \(i\). See sample_gaussian_bins() for usage.