norse.torch.functional.regularization.regularize_step

norse.torch.functional.regularization.regularize_step(z: torch.Tensor, s: typing.Any, accumulator: Accumulator = <function spike_accumulator>, state: typing.Optional[typing.Any] = None) → Tuple[torch.Tensor, Any]

Takes one step for a regularizer that aggregates some information (based on the spike_accumulator function), which is pushed forward and returned for future inclusion in an error term.

Parameters:

z (torch.Tensor): Spikes from a cell s (Any): Neuron state accumulator (Accumulator): Accumulator that decides what should be accumulated state (Optional[Any]): The regularization state to be aggregated. Typically some numerical value like a count. Defaults to None

Return:

A tuple of (spikes, regularizer state)