update doc, upgrade reset_state, update projection models (#592)

brainpy/_src/dnn/interoperation_flax.py

@@ -86,7 +86,7 @@ def initialize_carry(self, rng, batch_dims, size=None, init_fn=None):
         raise NotImplementedError
 
       _state_vars = self.model.vars().unique().not_subset(bm.TrainVar)
-      self.model.reset_state(batch_size=batch_dims)
+      self.model.reset(batch_size=batch_dims)
       return [_state_vars.dict(), 0, 0.]
 
     def setup(self):

brainpy/_src/dyn/projections/align_post.py

@@ -141,6 +141,10 @@ def update(self, x):
     self.refs['syn'].add_current(current)  # synapse post current
     return current
 
+  syn = property(lambda self: self.refs['syn'])
+  out = property(lambda self: self.refs['out'])
+  post = property(lambda self: self.refs['post'])
+
 
 class FullProjAlignPostMg(Projection):
   """Full-chain synaptic projection with the align-post reduction and the automatic synapse merging.
@@ -270,6 +274,12 @@ def update(self):
     self.refs['syn'].add_current(current)  # synapse post current
     return current
 
+  syn = property(lambda self: self.refs['syn'])
+  out = property(lambda self: self.refs['out'])
+  delay = property(lambda self: self.refs['delay'])
+  pre = property(lambda self: self.refs['pre'])
+  post = property(lambda self: self.refs['post'])
+
 
 class HalfProjAlignPost(Projection):
   """Defining the half-part of synaptic projection with the align-post reduction.
@@ -363,6 +373,8 @@ def update(self, x):
     self.refs['out'].bind_cond(g)  # synapse post current
     return current
 
+  post = property(lambda self: self.refs['post'])
+
 
 class FullProjAlignPost(Projection):
   """Full-chain synaptic projection with the align-post reduction.
@@ -488,3 +500,8 @@ def update(self):
     g = self.syn(self.comm(x))
     self.refs['out'].bind_cond(g)  # synapse post current
     return g
+
+  delay = property(lambda self: self.refs['delay'])
+  pre = property(lambda self: self.refs['pre'])
+  post = property(lambda self: self.refs['post'])
+  out = property(lambda self: self.refs['out'])

brainpy/_src/dyn/projections/align_pre.py

@@ -195,6 +195,12 @@ def update(self, x=None):
     self.refs['out'].bind_cond(current)
     return current
 
+  pre = property(lambda self: self.refs['pre'])
+  post = property(lambda self: self.refs['post'])
+  syn = property(lambda self: self.refs['syn'])
+  delay = property(lambda self: self.refs['delay'])
+  out = property(lambda self: self.refs['out'])
+
 
 class FullProjAlignPreDSMg(Projection):
   """Full-chain synaptic projection with the align-pre reduction and delay+synapse updating and merging.
@@ -326,6 +332,11 @@ def update(self):
     self.refs['out'].bind_cond(current)
     return current
 
+  pre = property(lambda self: self.refs['pre'])
+  post = property(lambda self: self.refs['post'])
+  syn = property(lambda self: self.refs['syn'])
+  out = property(lambda self: self.refs['out'])
+
 
 class FullProjAlignPreSD(Projection):
   """Full-chain synaptic projection with the align-pre reduction and synapse+delay updating.
@@ -454,6 +465,12 @@ def update(self, x=None):
     self.refs['out'].bind_cond(current)
     return current
 
+  pre = property(lambda self: self.refs['pre'])
+  post = property(lambda self: self.refs['post'])
+  syn = property(lambda self: self.refs['syn'])
+  delay = property(lambda self: self.refs['delay'])
+  out = property(lambda self: self.refs['out'])
+
 
 class FullProjAlignPreDS(Projection):
   """Full-chain synaptic projection with the align-pre reduction and delay+synapse updating.
@@ -581,3 +598,9 @@ def update(self):
     g = self.comm(self.syn(spk))
     self.refs['out'].bind_cond(g)
     return g
+
+  pre = property(lambda self: self.refs['pre'])
+  post = property(lambda self: self.refs['post'])
+  delay = property(lambda self: self.refs['delay'])
+  out = property(lambda self: self.refs['out'])
+

brainpy/_src/dyn/projections/plasticity.py

@@ -189,6 +189,12 @@ def __init__(
     self.A1 = A1
     self.A2 = A2
 
+  pre = property(lambda self: self.refs['pre'])
+  post = property(lambda self: self.refs['post'])
+  syn = property(lambda self: self.refs['syn'])
+  delay = property(lambda self: self.refs['delay'])
+  out = property(lambda self: self.refs['out'])
+
   def update(self):
     # pre-synaptic spikes
     pre_spike = self.refs['delay'].at(self.name)  # spike