Pytorch 1.0.0版本问题之module ‘torch.nn‘ has no attribute ‘Flatten‘

1. 原因：pytorch1.0.0没有Flatten类
2. 解决方法（有效！！，请用pytorch1.9.0下的文件进行替换）

pytorch1.9.0链接：

1. 第一步，找到本地安装pytorch的路径，我是通过ANACONDA安装，路径如下

找到以下路径

对其中的__init__文件修改

from .flatten import Flatten, Unflatten

2.第二步，替换文件

解压找到目录D:xx\pytorch-1.9.0\pytorch-1.9.0\torch\nn\modules中的flatten.py文件复制到自己电脑安装的相同路径下

3.第三步，替换第二个文件

找到你下载的1.9.0的文件下的D:xx\pytorch-1.9.0\torch的types.py文件

也放到原来的同样的位置

4.最后，修改刚刚复制过来的types.py文件的第21行，注释掉这行