Creating Tensors with different functions in Tensorflow