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