在 ‘__init__.pyi‘ 中找不到引用 ‘Flatten‘

核心问题：torch包版本过低

在复现fixmatch时，想要改写网络架构，用到了Flatten，发现函数黄体高亮，显示“在 '__init__.pyi' 中找不到引用 'Flatten' ”

找了几个方法试了一下，最终发现是环境问题，此环境问题往往难以发现，故记录之。

torch==1.5.0（gpu）

代码块：

torch==2.4.0（gpu）

没问题了。