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)