Merge pull request #5 from stefanopini/master

Fix bug when batch_first is False

Author: Andrea Palazzi

convlstm.py

@@ -114,7 +114,7 @@ def forward(self, input_tensor, hidden_state=None):
         """
         if not self.batch_first:
             # (t, b, c, h, w) -> (b, t, c, h, w)
-            input_tensor.permute(1, 0, 2, 3, 4)
+            input_tensor = input_tensor.permute(1, 0, 2, 3, 4)
 
         # Implement stateful ConvLSTM
         if hidden_state is not None: