fix bug (#105)

Co-authored-by: bamsumit <bam_sumit@hotmail.com>

src/lava/lib/dl/slayer/block/base.py

@@ -94,7 +94,7 @@ def forward(self, x):
             if self.neuron is not None:
                 self.input_shape = self.neuron.shape
             else:
-                self.input_shape = input.shape[1:-1]
+                self.input_shape = x.shape[1:-1]
 
         if self.count_log is True:
             return x, torch.mean(x > 0)