netket.nn.to_array#

netket.nn.to_array(hilbert, apply_fun, variables, normalize=True, allgather=True)[source]#
Computes apply_fun(variables, states) on all states of hilbert and returns

the results as a vector.

Parameters

  • normalize – If True, the vector is normalized to have L2-norm 1.

  • allgather – If True, the final wave function is stored in full at all MPI ranks.