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)\).