Turn Numpy Array Into Tensorflow Vs Torch