netket.jax.expect#

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

Computes the expectation value over a log-pdf.

Parameters

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

  • expected_ffun –

  • expected_fun (Callable[[Any, jax._src.basearray.Array], jax._src.basearray.Array]) –

  • pars (Any) –

  • σ (jax._src.basearray.Array) –

  • n_chains (Optional[int]) –

Return type

Tuple[Array, Stats]