Add reshape shortcut for ReshapeLayer

Lasagne · Feb 7, 2015 · cd34882 · cd34882
1 parent 461d992
commit cd34882
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/lasagne/layers/shape.py b/lasagne/layers/shape.py
@@ -11,6 +11,7 @@
     "FlattenLayer",
     "flatten",
     "ReshapeLayer",
+    "reshape",
     "PadLayer",
     "pad",
 ]
@@ -134,6 +135,8 @@ def get_output_for(self, input, *args, **kwargs):
         # Everything else is handled by Theano
         return input.reshape(tuple(output_shape))
 
+reshape = ReshapeLayer # shortcut
+
 
 class PadLayer(Layer):
     def __init__(self, incoming, width, val=0, batch_ndim=2, **kwargs):
@@ -155,4 +158,4 @@ def get_output_shape_for(self, input_shape):
     def get_output_for(self, input, *args, **kwargs):
         return padding.pad(input, self.width, self.val, self.batch_ndim)
 
-pad = PadLayer # shortcut
+pad = PadLayer # shortcut