return empty array if monitor is empty. network/monitors.py

Author: Hananel-Hazan

bindsnet/network/monitors.py

@@ -54,6 +54,8 @@ def __init__(
         if self.time is None:
             self.device = "cpu"
 
+        self.clean = True
+
         self.recording = []
         self.reset_state_variables()
 
@@ -68,16 +70,20 @@ def get(self, var: str) -> torch.Tensor:
         Note, if time == `None`, get return the logs and empty the monitor variable
 
         """
-        return_logs = torch.cat(self.recording[var], 0)
-        if self.time is None:
-            self.recording[var] = []
+        if self.clean:
+            return_logs = torch.empty(0, device=self.device)
+        else:
+            return_logs = torch.cat(self.recording[var], 0)
+            if self.time is None:
+                self.recording[var] = []
         return return_logs
 
     def record(self) -> None:
         # language=rst
         """
         Appends the current value of the recorded state variables to the recording.
         """
+        self.clean = False
         for v in self.state_vars:
             data = getattr(self.obj, v).unsqueeze(0)
             # self.recording[v].append(data.detach().clone().to(self.device))
@@ -101,6 +107,7 @@ def reset_state_variables(self) -> None:
             self.recording = {
                 v: [[] for i in range(self.time)] for v in self.state_vars
             }
+        self.clean = True
 
 
 class NetworkMonitor(AbstractMonitor):