remove the reshape after input

This commit is contained in:
jeanfad 2016-04-06 15:57:15 +02:00
Родитель d14daf45f2
Коммит 829d2d203a
1 изменённых файлов: 1 добавлений и 5 удалений

Просмотреть файл

@ -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