Tensorflow Simple Code For Python