netket.experimental.models.GRUNet1D#

class netket.experimental.models.GRUNet1D[source]#

Bases: RNN

Gated recurrent unit network. Only supports one previous neighbor at each site.

Attributes
layers: int#

number of layers.

features: Union[Iterable[int], int]#

output feature density in each layer. If a single number is given, all layers except the last one will have the same number of features.

hilbert: HomogeneousHilbert#

the Hilbert space. Only homogeneous unconstrained Hilbert spaces are supported.

Methods
__call__(inputs)#

Computes the log wave-functions for input configurations.

Parameters:

inputs (Union[ndarray, Array]) – configurations with dimensions (batch, Hilbert.size).

Return type:

Union[ndarray, Array]

Returns:

The log psi with dimension (batch,).