Tensorflow 2.0 Input Data