From 55447cbb3d33c2c209fdad8a1d53a166bc39544e Mon Sep 17 00:00:00 2001 From: Fariz Rahman Date: Wed, 10 Aug 2016 02:29:47 +0530 Subject: [PATCH] Bug fix : squeeze (#3433) --- keras/backend/theano_backend.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/keras/backend/theano_backend.py b/keras/backend/theano_backend.py index 7afe26261..eade21f5d 100644 --- a/keras/backend/theano_backend.py +++ b/keras/backend/theano_backend.py @@ -504,11 +504,9 @@ def expand_dims(x, dim=-1): def squeeze(x, axis): '''Remove a 1-dimension from the tensor at index "axis". ''' - broadcastable = x.broadcastable[:axis] + x.broadcastable[axis+1:] - x = T.patternbroadcast(x, [i == axis for i in range(x.type.ndim)]) - x = T.squeeze(x) - x = T.patternbroadcast(x, broadcastable) - return x + shape = list(x.shape) + shape.pop(axis) + return T.reshape(x, tuple(shape)) def temporal_padding(x, padding=1):