tf.shape() get wrong shape in tensorflow tf.shape() get wrong shape in tensorflow python-3.x python-3.x