ProjAlignPreMg2#

class brainpy.dyn.ProjAlignPreMg2(pre, delay, syn, comm, out, post, out_label=None, name=None, mode=None)[source]#

Synaptic projection which defines the synaptic computation with the dimension of presynaptic neuron group.

To simulate an E/I balanced network model:

class EINet(bp.DynSysGroup):
  def __init__(self):
    super().__init__()
    ne, ni = 3200, 800
    self.E = bp.dyn.LifRef(ne, V_rest=-60., V_th=-50., V_reset=-60., tau=20., tau_ref=5.,
                           V_initializer=bp.init.Normal(-55., 2.))
    self.I = bp.dyn.LifRef(ni, V_rest=-60., V_th=-50., V_reset=-60., tau=20., tau_ref=5.,
                           V_initializer=bp.init.Normal(-55., 2.))
    self.E2E = bp.dyn.ProjAlignPreMg2(pre=self.E,
                                      delay=0.1,
                                      syn=bp.dyn.Expon.desc(size=ne, tau=5.),
                                      comm=bp.dnn.JitFPHomoLinear(ne, ne, prob=0.02, weight=0.6),
                                      out=bp.dyn.COBA(E=0.),
                                      post=self.E)
    self.E2I = bp.dyn.ProjAlignPreMg2(pre=self.E,
                                      delay=0.1,
                                      syn=bp.dyn.Expon.desc(size=ne, tau=5.),
                                      comm=bp.dnn.JitFPHomoLinear(ne, ni, prob=0.02, weight=0.6),
                                      out=bp.dyn.COBA(E=0.),
                                      post=self.I)
    self.I2E = bp.dyn.ProjAlignPreMg2(pre=self.I,
                                      delay=0.1,
                                      syn=bp.dyn.Expon.desc(size=ni, tau=10.),
                                      comm=bp.dnn.JitFPHomoLinear(ni, ne, prob=0.02, weight=6.7),
                                      out=bp.dyn.COBA(E=-80.),
                                      post=self.E)
    self.I2I = bp.dyn.ProjAlignPreMg2(pre=self.I,
                                      delay=0.1,
                                      syn=bp.dyn.Expon.desc(size=ni, tau=10.),
                                      comm=bp.dnn.JitFPHomoLinear(ni, ni, prob=0.02, weight=6.7),
                                      out=bp.dyn.COBA(E=-80.),
                                      post=self.I)

  def update(self, inp):
    self.E2E()
    self.E2I()
    self.I2E()
    self.I2I()
    self.E(inp)
    self.I(inp)
    return self.E.spike

model = EINet()
indices = bm.arange(1000)
spks = bm.for_loop(lambda i: model.step_run(i, 20.), indices)
bp.visualize.raster_plot(indices, spks, show=True)
Parameters:
update()[source]#

The function to specify the updating rule.