Zerro out

Author: n-shevko

bindsnet/network/topology.py

@@ -209,7 +209,15 @@ def update(self, **kwargs) -> None:
 
         :param bool learning: Whether to allow connection updates.
         """
-        pass
+        mask = kwargs.get("mask", None)
+        if mask is None:
+            return
+
+        for f in self.pipeline:
+            if type(f).__name__ != 'Weight':
+                continue
+
+            f.value.masked_fill_(mask, 0)
 
     @abstractmethod
     def reset_state_variables(self) -> None:
@@ -484,6 +492,8 @@ def update(self, **kwargs) -> None:
         """
         learning = kwargs.get("learning", False)
         if learning and not self.manual_update:
+            super().update(**kwargs)
+
             # Pipeline learning
             for f in self.pipeline:
                 f.update(**kwargs)

examples/dosidicus/network.py

@@ -3,7 +3,7 @@
 from bindsnet.network import Network
 from bindsnet.network.nodes import LIFNodes
 from bindsnet.network.topology import MulticompartmentConnection
-from bindsnet.network.topology_features import Weight, Bias, Mask
+from bindsnet.network.topology_features import Weight, Bias
 from bindsnet.learning.MCC_learning import PostPre
 
 network = Network(dt=1.0)
@@ -22,17 +22,22 @@
 )
 bias = Bias(name='bias_feature', value=torch.rand(5, 5))
 
-mask = torch.tril(torch.ones((5, 5)), diagonal=-1).bool()
-
 connection = MulticompartmentConnection(
     source=source_layer,
     target=target_layer,
-    pipeline=[weight, Mask(name='mask', value=mask), bias],
+    pipeline=[weight, bias],
     device='cpu'
 )
 network.add_connection(connection, source="input", target="output")
 print(connection.pipeline[0].value)
 network.run(
-    inputs={"input": torch.bernoulli(torch.rand(250, 5)).byte()}, time=250
+    inputs={"input": torch.bernoulli(torch.rand(250, 5)).byte()},
+    time=250,
+    masks={
+       ('input', 'output'): ~torch.tril(torch.ones((5, 5)), diagonal=-1).bool()
+    }
 )
+
+print(network.layers['input'].v)
+print(network.layers['output'].v)
 print(connection.pipeline[0].value)