add not self.inputs.targeted and to required components. Fixing general __repr__ bug in jaxcomponent

Author: rxng8

ngclearn/components/input_encoders/bernoulliCell.py

@@ -48,7 +48,7 @@ def advance_state(self, t):
     @compilable
     def reset(self):
         restVals = jnp.zeros((self.batch_size.get(), self.n_units.get()))
-        self.inputs.set(restVals)
+        not self.inputs.targeted and self.inputs.set(restVals)
         self.outputs.set(restVals)
         self.tols.set(restVals)
 

ngclearn/components/input_encoders/latencyCell.py

@@ -211,7 +211,7 @@ def advance_state(self, t):
     @compilable
     def reset(self):
         restVals = jnp.zeros((self.batch_size.get(), self.n_units.get()))
-        self.inputs.set(restVals)
+        not self.inputs.targeted and self.inputs.set(restVals)
         self.outputs.set(restVals)
         self.tols.set(restVals)
         self.mask.set(restVals)

ngclearn/components/input_encoders/phasorCell.py

@@ -88,7 +88,7 @@ def advance_state(self, t, dt):
     @compilable
     def reset(self):
         restVals = jnp.zeros((self.batch_size.get(), self.n_units.get()))
-        self.inputs.set(restVals)
+        not self.inputs.targeted and self.inputs.set(restVals)
         self.outputs.set(restVals)
         self.tols.set(restVals)
         self.angles.set(restVals)

ngclearn/components/jaxComponent.py

@@ -63,7 +63,7 @@ def __repr__(self):
         maxlen = max(len(c) for c in comps) + 5
         lines = f"[{self.__class__.__name__}] PATH: {self.name}\n"
         for c in comps:
-            stats = tensorstats(getattr(self, c).value)
+            stats = tensorstats(getattr(self, c).get())
             if stats is not None:
                 line = [f"{k}: {v}" for k, v in stats.items()]
                 line = ", ".join(line)

ngclearn/components/other/expKernel.py

@@ -86,7 +86,7 @@ def advance_state(self, t):
     def reset(self):
         restVals = jnp.zeros((self.batch_size, self.n_units)) ## inputs, epsp
         restTensor = jnp.zeros([self.win_len, self.batch_size, self.n_units], jnp.float32)  ## tf
-        self.inputs.set(restVals)
+        not self.inputs.targeted and self.inputs.set(restVals)
         self.epsp.set(restVals)
         self.tf.set(restTensor)
 

ngclearn/components/other/varTrace.py

@@ -1,3 +1,5 @@
+# %%
+
 from ngclearn.components.jaxComponent import JaxComponent
 from jax import numpy as jnp, random, jit
 from functools import partial
@@ -124,7 +126,7 @@ def advance_state(self, dt):
     @compilable
     def reset(self):
         restVals = jnp.zeros((self.batch_size, self.n_units))
-        self.inputs.set(restVals)
+        not self.inputs.targeted and self.inputs.set(restVals)
         self.outputs.set(restVals)
         self.trace.set(restVals)
 

ngclearn/components/synapses/hebbian/hebbianSynapse.py

@@ -257,7 +257,7 @@ def evolve(self):
     def reset(self, batch_size, shape):
         preVals = jnp.zeros((batch_size, shape[0]))
         postVals = jnp.zeros((batch_size, shape[1]))
-        self.inputs.set(preVals) # inputs
+        not self.inputs.targeted and self.inputs.set(preVals) # inputs
         self.outputs.set(postVals) # outputs
         self.pre.set(preVals) # pre
         self.post.set(postVals) # post

ngclearn/components/synapses/modulated/REINFORCESynapse.py

@@ -207,7 +207,7 @@ def reset(self, batch_size, shape):
         seed = jax.random.PRNGKey(42)
 
 
-        self.inputs.set(inputs)
+        not self.inputs.targeted and self.inputs.set(inputs)
         self.outputs.set(outputs)
         self.objective.set(objective)
         self.rewards.set(rewards)

ngclearn/components/synapses/patched/hebbianPatchedSynapse.py

@@ -280,7 +280,7 @@ def evolve(self):
     def reset(self, batch_size, shape):
         preVals = jnp.zeros((batch_size, shape[0]))
         postVals = jnp.zeros((batch_size, shape[1]))
-        self.inputs.set(preVals) # inputs
+        not self.inputs.targeted and self.inputs.set(preVals) # inputs
         self.outputs.set(postVals) # outputs
         self.pre.set(preVals) # pre
         self.post.set(postVals) # post

ngclearn/components/synapses/patched/patchedSynapse.py

@@ -159,7 +159,7 @@ def reset(self, batch_size, shape):
         postVals = jnp.zeros((batch_size, shape[1]))
         inputs = preVals
         outputs = postVals
-        self.inputs.set(inputs)
+        not self.inputs.targeted and self.inputs.set(inputs)
         self.outputs.set(outputs)
 
     def save(self, directory, **kwargs):