解决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)

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

总结

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

您可能感兴趣的文章:

python requests post的使用方式
这篇文章主要介绍了python requests post的使用方式，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教
2023-03-03
python中geopandas库安装出现各种问题的解决办法
这篇文章主要介绍了关于python中geopandas库安装出现各种问题的解决办法,总结了在Windows下两种安装geopandas库的方法,方法一是在新环境下使用conda命令安装,方法二通过离线安装GDAL、Fiona、Pyproj、Rtree、Shapely五个库,再用pip安装geopandas,需要的朋友可以参考下
2024-11-11
基于Python实现成语填空游戏的示例代码
成语填空想必大家都是十分熟悉的了，特别是有在上小学的家长肯定都有十分深刻的印象。当然了你也别小看了成语调控小游戏，有的时候知识储备不够，你还真的不一定猜得出来是什么。本文就来用Python编写一个简单的成语填空游戏，感兴趣的可以了解下
2023-02-02
python 切片和range()用法说明
首先需要明白，可迭代对象，按照正数索引（正序）是从0开始的，按照负数索引（逆序）是从－1开始的
2013-03-03
Python控制台输出俄罗斯方块的方法实例
这篇文章主要给大家介绍了关于Python控制台输出俄罗斯方块的相关资料,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2021-04-04
解决django接口无法通过ip进行访问的问题
这篇文章主要介绍了解决django接口无法通过ip进行访问的问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2020-03-03
python Tkinter实例详解
tkinter(Tk interface)是Python的标准GUl库，支持跨平台的GUl程序开发。tkinter适合小型的GUl程序编写，也特别适合初学者学习GUl编程，这篇文章主要介绍了python Tkinter详解,需要的朋友可以参考下
2023-03-03
如何使用python编写一个简单的课时记录系统
编写一个应用系统需要多方面的知识和技能,下面这篇文章主要给大家介绍了关于如何使用python编写一个简单的课时记录系统的相关资料,文中通过代码介绍的非常详细,对大家学习或者使用python具有一定的参考借鉴价值,需要的朋友可以参考下
2024-04-04
如何使用yolov5输出检测到的目标坐标信息
YOLOv5是一系列在 COCO 数据集上预训练的对象检测架构和模型,下面这篇文章主要给大家介绍了关于如何使用yolov5输出检测到的目标坐标信息的相关资料,需要的朋友可以参考下
2022-03-03
Python字符串详细介绍
这篇文章主要介绍了Python字符串详解,本文讲解了字符串相关知识、字符串的一些特性、原始字符串、unicode字符串、字符串的常用操作方法、内建函数列表等内容,需要的朋友可以参考下
2015-05-05