rockpool.nn.modules.torch.lif_torch

Implement a LIF Module, using a Torch backend

Provides LIFBaseTorch base class for LIF torch modules, and LIFTorch module.

Functions

bitshift_to_decay(*dashes)
bitshift_to_tau(dt, *dashes)
decay_to_bitshift(*decays)
decay_to_tau(dt, *decays)
sigmoid(x, threshold)	Sigmoid function
tau_to_bitshift(dt, *taus)
tau_to_decay(dt, *taus)

Classes

LIFBaseTorch(*args, **kwargs)
LIFTorch(*args, **kwargs)	A leaky integrate-and-fire spiking neuron model with a Torch backend
PeriodicExponential(*args, **kwargs)	Subtract from membrane potential on reaching threshold
StepPWL(*args, **kwargs)	Heaviside step function with piece-wise linear surrogate to use as spike-generation surrogate