quantax.model.RBM_Dense

quantax.model.RBM_Dense(features: int, use_bias: bool = True, dtype: ~numpy.dtype = <class 'jax.numpy.float32'>)

The restricted Boltzmann machine with one dense layer \(\psi(s) = \prod \cosh(W s + b)\).

Parameters:

• features – The number of output features or hidden units.

• use_bias – Whether to add on a bias.

• dtype – The data type of the parameters.