Pytorch实现softmax回归

1.导入相关包

#【1】导入相关包
import torch
from torch import nn
from torch.nn import init
import torchvision
import torchvision.transforms as transforms
import matplotlib.pyplot as plt
import time
import numpy as np
import sys
sys.path.append("..")#导入本地文件。两个点代表上一层目录。
import my_pytorch as mp

2.获取和读取数据

#【2】获取和读取数据

#【2.1】定义获取label对应的值，及可视化数据（图像）的方法
def get_fashion_mnist_labels(labels):
	text_labels=['t-shirt', 'trouser', 'pullover', 'dress', 'coat',
                   'sandal', 'shirt', 'sneaker', 'bag', 'ankle boot']
    return [text_labels[int(i)] for i in labels]#labels是0-9的数字组成的list

def show_fashion_mnist(images,labels):
	#mp.use_svg_display()
	_,figs=plt.subplots(1,len(images),figsize=(12,12))
	for f,img,lbl in zip(figs,images,labels):
		f.imshow(img.view((28,28)).numpy())#img现在是tensor，要转numpy数组
		f.set_title(lbl)
		f.axes.get_xaxis().set_visible(False)
		f.axes.get_yaxis().set_visible(False)
	plt.show()
	
#【2.2】获取、显示、读取数据
	
batch_size=256

def load_data_fashion_mnist():
	#【2.2.1】使用torchvision.datasets模块下载数据集
	mnist_train=torchvision.datasets.FashionMNIST(root='~/Datasets/FashionMNIST',train=False,download=True,transform=transforms.ToTensor(