屎山代码样例,游戏一生一次传染病代码

class Flatten(nn.Module): def __init__(self): super(Flatten, self).__init__() def forward(self, input): return input.view(input.size(0), -1)