norse.torch.functional.regularization.regularize_step

norse.torch.functional.regularization.regularize_step#

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

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)