Numpy Array Vs Pytorch Tensor