实验2任务3code完成

2023-10-24 16:35:36 +08:00
parent bef19fd9f0
commit 52ef87ec87
8 changed files with 174 additions and 8 deletions
--- a/Lab2/code/2.2.py
+++ b/Lab2/code/2.2.py
@@ -1,7 +1,7 @@
 import time
 import numpy as np
 import torch
-from torch.nn.functional import one_hot, softmax
+from torch.nn.functional import *
 from torch.utils.data import Dataset, DataLoader
 from torch import nn
 from torchvision import datasets, transforms