jaxsnn.base.dataset.dataloader

Classes

Any(*args, **kwargs)	Special type indicating an unconstrained type.
EventPropSpike(time, idx, current)

Functions

jaxsnn.base.dataset.dataloader.data_loader(dataset: Tuple[Any, Any], batch_size: int, num_batches: Optional[int, None] = None, rng: Optional[jax.Array, None] = None)