netket.models.RBMSymm

class netket.models.RBMSymm

Bases: Module

A symmetrized RBM using the netket.nn.DenseSymm() layer internally.

Attributes

alpha: float | int = 1

feature density. Number of features equal to alpha * input.shape[-1]

precision: Any = None

Numerical precision of the computation see jax.lax.Precision for details.

use_hidden_bias: bool = True

if True uses a bias in the dense layer (hidden layer bias).

use_visible_bias: bool = True

if True adds a bias to the input not passed through the nonlinear layer.

symmetries: HashableArray | PermutationGroup

A group of symmetry operations (or array of permutation indices) over which the layer should be invariant. Numpy/Jax arrays must be wrapped into an netket.utils.HashableArray.

activation: Any

The nonlinear activation function.

kernel_init: Callable[[Any, Sequence[int], None | str | type[Any] | dtype | _SupportsDType], Array]

Initializer for the Dense layer matrix.

hidden_bias_init: Callable[[Any, Sequence[int], None | str | type[Any] | dtype | _SupportsDType], Array]

Initializer for the hidden bias.

visible_bias_init: Callable[[Any, Sequence[int], None | str | type[Any] | dtype | _SupportsDType], Array]

Initializer for the visible bias.

Methods

__call__(x_in)

Call self as a function.