文章目录 前言 代码实现 1、导入数据库 关于torch.manual_seed(1) 2、参数设置 3、数据导入 4、定义MMD损失 5、定义训练函数 5.1 nn.CrossEntropyLoss() 5.2 .detach() 5.3 .size VS .shape 5.4 .to(DEVICE) 5.5 .max() 5.6 optimizer.zero_grad()