jaxsnn.base.dataset.dataloader

Classes

EventPropSpike(time, idx, current)

Functions

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