Use numpy's concatenate instead of hstack, it's apparently faster

brian-team · Jul 23, 2013 · ac84555 · ac84555
1 parent 1f2642c
commit ac84555
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/brian2/synapses/synapses.py b/brian2/synapses/synapses.py
@@ -194,8 +194,8 @@ def pre_update(self):
         # Push new spikes into the queue
         spikes = self.source.spikes
         if len(spikes):
-            indices = np.hstack((self.synapse_indices[spike]
-                                 for spike in spikes)).astype(np.int32)
+            indices = np.concatenate([self.synapse_indices[spike]
+                                      for spike in spikes]).astype(np.int32)
             if len(indices):
                 if len(self._delays) > 1:
                     delays = np.round(self._delays[indices] / self.dt).astype(int)