李沐深度学习-多层感知机简洁实现

import torch
import numpy as np
import torch.nn as nn
from torch.nn import init
import torchvision.transforms as transforms
import torch.utils.data as Data
import sys

sys.path.append("路径")
import d2lzh_pytorch as d2l

num_inputs, num_outputs, num_hidden = 784, 10, 256


# X形状转换
class FlattenLayer(nn.Module):
    def __init__(self):
        super(FlattenLayer, self).__init__()

    def forward(self, X):
        return