From 829d2d203adeb49dfefc54324d97ae55f55d3a48 Mon Sep 17 00:00:00 2001
From: jeanfad <jeanfad@microsoft.com>
Date: Wed, 6 Apr 2016 15:57:15 +0200
Subject: [PATCH] remove the reshape after input

---
 contrib/Python/cntk/graph.py | 6 +-----
 1 file changed, 1 insertion(+), 5 deletions(-)

diff --git a/contrib/Python/cntk/graph.py b/contrib/Python/cntk/graph.py
index e5577e924..95480e75c 100644
--- a/contrib/Python/cntk/graph.py
+++ b/contrib/Python/cntk/graph.py
@@ -496,14 +496,10 @@ def _get_input_node(list_of_tensors, has_sequence_dimension, **kw):
     # followed by a reshape node that has the dims '2:3'. So we have 2*3 = 6 
     # dimensions when flattened out for the reader. 
     dims = int(np.multiply.reduce(cntk_shape))
-    node = cntk1_ops.Input(dims, **kw)
+    node = cntk1_ops.Input(cntk_shape, **kw)
     node.reader = CNTKTextFormatReader(tf.name)
     node.reader.add_input(node, alias, dims)
     
-    if len(cntk_shape) > 1:
-        node = cntk1_ops.NewReshape(node,
-                dims=cntk_shape)
-
     return node