Tensorflow Convert Numpy Array To Pytorch