make lasagne.utils.concatenate support negative axes.

Lasagne · Jan 30, 2015 · a53867d · a53867d
1 parent 5a7ff66
commit a53867d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/lasagne/utils.py b/lasagne/utils.py
@@ -91,6 +91,9 @@ def concatenate(tensor_list, axis=0):
         - out : tensor
             the concatenated tensor expression.
     """
+    if axis < 0:
+        axis += tensor_list[0].ndim
+
     concat_size = sum(tensor.shape[axis] for tensor in tensor_list)
 
     output_shape = ()