SuperSpike

class src.model.SurrogateGradient.SuperSpike(*args, **kwargs)

Bases: Function

Class that implements the SuperSpike function as the surrogate gradient for the backward pass.

static backward(ctx, grad_output: Tensor) → Tensor

Backward pass using surrogate gradient function: SuperSpike

static forward(ctx, input_: Tensor) → Tensor

Forward pass using 'Heaviside step function'

spiking_mode: str = 'binary'

surrogate_scale: float = 10.0