quantax.model.SingleDense

quantax.model.SingleDense(features: int, actfn: Callable, use_bias: bool = True, holomorphic: bool = False, dtype: str | type[Any] | dtype | SupportsDType = <class 'jax.numpy.float32'>)

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