Pytorch Transform Numpy To Tensor