Save A Pytorch Tensor