Merge pull request #66 from Hananel-Hazan/hananel

Slight improve performance

Author: Dan Saunders

bindsnet/network/nodes.py

@@ -195,7 +195,7 @@ def _reset(self):
 		Resets relevant state variables.
 		'''
 		super()._reset()
-		self.v[self.v != self.rest] = self.rest        # Neuron voltages.
+		self.v = torch.zeros(self.v.size()) + self.rest         # Neuron voltages.
 		self.refrac_count[self.refrac_count != 0] = 0  # Refractory period counters.
 
 
@@ -228,7 +228,7 @@ def __init__(self, n=None, shape=None, traces=False, thresh=-52.0, rest=-65.0,
 		self.refrac = refrac  # Post-spike refractory period.
 		self.decay = decay    # Rate of decay of neuron voltage.
 
-		self.v = self.rest * torch.ones(self.shape)  # Neuron voltages.
+		self.v = torch.zeros(self.v.size()) + self.rest         # Neuron voltages.
 		self.refrac_count = torch.zeros(self.shape)  # Refractory period counters.
 
 	def step(self, inpts, dt):
@@ -263,7 +263,7 @@ def _reset(self):
 		Resets relevant state variables.
 		'''
 		super()._reset()
-		self.v[self.v != self.rest] = self.rest        # Neuron voltages.
+		self.v = torch.zeros(self.v.size()) + self.rest         # Neuron voltages.
 		self.refrac_count[self.refrac_count != 0] = 0  # Refractory period counters.
 
 
@@ -339,7 +339,7 @@ def _reset(self):
 		Resets relevant state variables.
 		'''
 		super()._reset()
-		self.v[self.v != self.rest] = self.rest        # Neuron voltages.
+		self.v = torch.zeros(self.v.size()) + self.rest         # Neuron voltages.
 		self.refrac_count[self.refrac_count != 0] = 0  # Refractory period counters.
 
 
@@ -421,7 +421,7 @@ def _reset(self):
 		Resets relevant state variables.
 		'''
 		super()._reset()
-		self.v[self.v != self.rest] = self.rest        # Neuron voltages.
+		self.v = torch.zeros(self.v.size()) + self.rest        # Neuron voltages.
 		self.refrac_count[self.refrac_count != 0] = 0  # Refractory period counters.