Turn Numpy Array Into Pytorch Tensor