pytorch代码

import argparseimport osimport numpy as npimport mathimport scipyimport itertoolsimport mnistmimport torchvision.transforms as transformsfrom torchvision.utils import save_imagefrom torch.utils.data import DataLoaderfrom torchvision