jaxsnn.base.dataset.constant

Functions

jaxsnn.base.dataset.constant.constant_dataset(t_max: float, size: int) → Tuple[jax.Array, jax.Array]