解决ImportError:cannot import name ‘Flatten‘ from ‘torch.nn‘问题

更新时间：2023年06月15日 09:09:58 作者：全幼儿园最聪明

这篇文章主要介绍了解决ImportError:cannot import name ‘Flatten‘ from ‘torch.nn‘问题，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教

问题描述

在代码中从torch.nn引入Flatten报错：

解决方案

进入python交互式环境

输入以下代码，查看pytorch版本：

import torch
print(torch.__version__)

可以看到pytorch版本是1.1.0

版本太低不包含Flatten属性

因此，一个办法就是升级pytorch版本，但我们知道下载安装pytorch并不容易，若没有镜像下载缓慢且需要考虑显卡驱动、conda、torch三者的版本关系，升级可能导致版本不匹配conda无法使用gpu等问题。

另一个简单的办法是自己写Flatten类，这里我需要的功能是仅是扁平化，

参考代码如下：

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

引入后重新运行代码，没有报错

总结

以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

jupyter notebook 多环境conda kernel配置方式
这篇文章主要介绍了jupyter notebook 多环境conda kernel配置方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2020-04-04
基于Python OpenCV和 dlib实现眨眼检测
这篇文章主要介绍了基于Python OPenCV及dlib实现检测视频流中的眨眼次数。文中的代码对我们的学习和工作有一定价值，感兴趣的同学可以参考一下
2021-12-12
用Python实现一个简单的能够发送带附件的邮件程序的教程
这篇文章主要介绍了用Python实现一个简单的能够发送带附件的邮件程序的教程,用MIMEApplication模块来发送各种类型的文件,需要的朋友可以参考下
2015-04-04
Keras 数据增强ImageDataGenerator多输入多输出实例
这篇文章主要介绍了Keras 数据增强ImageDataGenerator多输入多输出实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2020-07-07
为Python的Tornado框架配置使用Jinja2模板引擎的方法
Jinja2是人气Web框架Flask中的内置模板引擎,而且与Django的模板引擎比较类似,这里我们就来看一下为Python的Tornado框架配置使用Jinja2模板引擎的方法
2016-06-06
Selenium向iframe富文本框输入内容过程图解
这篇文章主要介绍了Selenium向iframe富文本框输入内容过程图解,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
2020-04-04
pyspark操作MongoDB的方法步骤
这篇文章主要介绍了pyspark操作MongoDB的方法步骤，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧
2019-01-01
Python元组(Tuple)全部高级用法大全
这段文章详细讲解了Python元组的用法,从基础到高级技巧,涵盖创建、解包、嵌套、与列表字典集合的转换等,还强调了元组在表示固定数据、防止误改、作为字典键等方面的优势,适合Python新手理解和掌握,需要的朋友可以参考下
2026-05-05
Python中的sort()和sorted()用法示例解析
本文给大家介绍Python中list.sort()和sorted()的使用区别,详细介绍其参数功能及Timsort排序算法特性,涵盖自适应性、稳定性、时间复杂度,并提及该算法在Java、JavaScript等多语言中的应用,感兴趣的朋友跟随小编一起看看吧
2025-08-08
PyCharm Terminal终端命令行Shell设置方式
这篇文章主要介绍了PyCharm Terminal终端命令行Shell设置方式，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教
2023-01-01