Pytorch Load Tensor From File