总结一些模块 1 Flatten from torch import nn import torch class Flatten(nn.Module): def __init__(self): super(Flatten, self).__init__(</