首页 > TAG信息列表 > 0903Softmax
0903Softmax简洁实现
点击查看代码 import torch import torchvision from torch import nn from d2l import torch as d2l from torch.utils import data from torchvision.transforms import transforms batch_size = 256 train_iter, test_iter = d2l.load_data_fashion_mnist(batch_size) # tran