Tensorflow Convert Numpy Array To Torch