AttributeError: module 'torch.nn' has no attribute 'Flatten'

这个错误通常是因为使用了过时的[pytorch](https://geek.csdn.net/educolumn/1309512bb95bbd063a82fd0157e558d8?spm=1055.2569.3001.10083)[版本](https://geek.csdn.net/educolumn/04449e2eb7da35a9c72dce97dbf43e2a?spm=1055.2569.3001.10083)导致的。在较新的[版本](https://geek.csdn.net/educolumn/04449e2eb7da35a9c72dce97dbf43e2a?spm=1055.2569.3001.10083)中，将`torch.nn.Flatten`替换为`torch.flatten`可以[解决](https://geek.csdn.net/educolumn/03368addd5d267af1b4c6df01bb4af26?spm=1055.2569.3001.10083)此问题。请确保您的PyTorch[版本](https://geek.csdn.net/educolumn/04449e2eb7da35a9c72dce97dbf43e2a?spm=1055.2569.3001.10083)是最新的，或者使用以下代码进行替换： ```python import torch import torch.nn as nn # 替换 torch.nn.Flatten class Flatten(nn.Module): def forward(self, x): return x.view(x.size()[0], -1) 这样，您就可以使用`Flatten`类来展平输入张量了。