Turn Numpy Array Into Torch Tensor Dtype Pytorch