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: Any, batch_size: int, rng: Optional[jax.Array, None] = None)