netket.jax.expect

netket.jax.expect(log_pdf, expected_fun, pars, σ, *expected_fun_args, n_chains=None)

Computes the expectation value over a log-pdf.

Parameters

• log_pdf (Callable[[Any, ndarray], ndarray]) –

• expected_ffun –

• expected_fun (Callable[[Any, jax._src.numpy.ndarray.ndarray], jax._src.numpy.ndarray.ndarray]) –

• pars (Any) –

• σ (jax._src.numpy.ndarray.ndarray) –

• n_chains (Optional[int]) –

Return type

Tuple[ndarray, Stats]