Tensorflow Create Tensor Python