002.动手实现softmax回归(pytorch简洁版)

1. 相关操作可复用002.从零开始实现softMax回归(pytorch)
   中的代码

import torch
from torch import nn
from torch.nn import init
import numpy as np
import sys
sys.path.append("..")
import d2lzh_pytorch as d2l

batch_size = 256
train_iter, test_iter = d2l.load_data_fashion_mnist(batch_size,root='/Users/wydi/PycharmProjects/DeepLearning_with_LiMu/datasets/FashionMnist')


batch_size = 256
num_input = 784
num_output = 10