Creating random tensors with TensorFlow