解决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)引入后重新运行代码，没有报错
总结
以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
大家好，本篇文章主要讲的是python3基础之集合set详解，感兴趣的同学赶快来看一看吧，对你有帮助的话记得收藏一下，方便下次浏览2021-12-12
本篇文章主要介绍了Python paramiko模块的使用示例，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧2018-04-04
Matplotlib是一款极其强大的Python数据可视化库，这篇文章中，我们将深入讨论 Matplotlib 的一些高级特性，包括对象导向接口、自定义颜色映射和样式、动态图形等，感兴趣的小伙伴跟着小编一起来探讨吧2023-07-07
在本篇文章中小编给大家分享的是关于Python将string转换到float的实例方法以及相关知识点，需要的朋友们参考下。2019-07-07
这篇文章主要为大家介绍了python日期时间处理，具有一定的参考价值，感兴趣的小伙伴们可以参考一下，希望能够给你带来帮助2021-12-12
解决Python内层for循环如何break出外层的循环的问题
今天小编就为大家分享一篇解决Python内层for循环如何break出外层的循环的问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-06-06
本文主要介绍pymysql安装、操作流程、语法基础及封装操作数据库类,需要的朋友可以参考下，希望能对大家有所帮助，每日提升一点点，欢迎大家多多交流讨论2021-08-08
这篇文章主要介绍了Python中下划线的使用方法,是为python编程学习中的基本知识,需要的朋友可以参考下2015-03-03
这篇文章主要介绍了Python 文件数据读写的具体实现,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2020-01-01
这篇文章主要介绍了python使用range函数计算一组数和的方法,涉及Python中range函数的使用技巧,具有一定参考借鉴价值,需要的朋友可以参考下2015-05-05
最新评论