首页 > TAG信息列表 > 0902Softmax
0902Softmax回归从零开始
点击查看代码 import pylab import torch from IPython import display from d2l import torch as d2l # softmax回归的从零开始实现 batch_size = 256 train_iter, test_iter = d2l.load_data_fashion_mnist(batch_size) # 初始化模型参数 # 将展平每个图像，把它们看作长度为78