Pytorch Tensor Vs Numpy Array