Tensorflow Lstm Input Shape