Integrate and Fire (IAF)

:class:~sinabs.layers.IAF layer.

import torch
from utils import plot_evolution

import sinabs.layers as sl

const_current = torch.ones((1, 100, 1)) * 0.03
single_current = torch.zeros((1, 100, 1))
single_current[:, 0] = 0.1

iaf_neuron = sl.IAF(record_states=True)
plot_evolution(iaf_neuron, const_current)


iaf_neuron = sl.IAF(tau_syn=15.0, record_states=True)
plot_evolution(iaf_neuron, single_current)

---------------------------------------------------------------------------
ModuleNotFoundError                       Traceback (most recent call last)
Cell In[1], line 2
      1 import torch
----> 2 from utils import plot_evolution
      4 import sinabs.layers as sl
      6 const_current = torch.ones((1, 100, 1)) * 0.03

ModuleNotFoundError: No module named 'utils'