手写数字识别MNIST仅用全连接层Linear实现

    import torch
    from torch import nn
    from torch import optim
    import torchvision
    from matplotlib import pyplot as plt
    from torch.utils.data import DataLoader
    
    
    def one_hot(label,depth=10):
    out = torch.zeros(label.size(0),depth)
    idx = torch.LongTensor(label).view(-1,1)
    out.scatter_(dim = 1,index = idx,value = 1)
    return out
    
    def plot_curve(data):
    fig = plt.figure()
    plt.plot(range(len(data)),data,color = 'blue')
    plt.legend(['value'],loc = 'upper right')
    plt.xlabel('step')
    plt.ylabel('value')
    plt.show(