TensorFlow comparison with Numpy