quantax.model.SingleDense

quantax.model.SingleDense(features: int, actfn: ~typing.Callable, use_bias: bool = True, holomorphic: bool = False, dtype: ~numpy.dtype = <class 'jax.numpy.float32'>)

Network with one dense layer \(\psi(s) = \prod f(W s + b)\).

Parameters:

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

• actfn – The activation function applied after the dense layer.

• use_bias – Whether to add on a bias.

• holomorphic – Whether the whole network is complex holomorphic.

• dtype – The data type of the parameters.