netket.models.RBMSymm#

class netket.models.RBMSymm[source]#

Bases: flax.linen.module.Module

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

Attributes
alpha: Union[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.

variables#

Returns the variables in this module.

Return type

Mapping[str, Mapping[str, Any]]

symmetries: Union[netket.utils.HashableArray, netket.utils.group._permutation_group.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.

Methods
activation()#
has_rng(name)#

Returns true if a PRNGSequence with name name exists.

Return type

bool

Parameters

name (str) –

kernel_init(shape, dtype=<class 'jax.numpy.float64'>)#
put_variable(col, name, value)#

Sets the value of a Variable.

Parameters
  • col (str) – the variable collection.

  • name (str) – the name of the variable.

  • value (Any) – the new value of the variable.

Returns:

visible_bias_init(shape, dtype=<class 'jax.numpy.float64'>)#