解決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交互式環(huán)境

輸入以下代碼，查看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)

引入后重新運行代碼，沒有報錯

總結

以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家。

您可能感興趣的文章:

python requests post的使用方式
這篇文章主要介紹了python requests post的使用方式，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教
2023-03-03
python中geopandas庫安裝出現(xiàn)各種問題的解決辦法
這篇文章主要介紹了關于python中geopandas庫安裝出現(xiàn)各種問題的解決辦法,總結了在Windows下兩種安裝geopandas庫的方法,方法一是在新環(huán)境下使用conda命令安裝,方法二通過離線安裝GDAL、Fiona、Pyproj、Rtree、Shapely五個庫,再用pip安裝geopandas,需要的朋友可以參考下
2024-11-11
基于Python實現(xiàn)成語填空游戲的示例代碼
成語填空想必大家都是十分熟悉的了，特別是有在上小學的家長肯定都有十分深刻的印象。當然了你也別小看了成語調(diào)控小游戲，有的時候知識儲備不夠，你還真的不一定猜得出來是什么。本文就來用Python編寫一個簡單的成語填空游戲，感興趣的可以了解下
2023-02-02
python 切片和range()用法說明
首先需要明白，可迭代對象，按照正數(shù)索引（正序）是從0開始的，按照負數(shù)索引（逆序）是從－1開始的
2013-03-03
Python控制臺輸出俄羅斯方塊的方法實例
這篇文章主要給大家介紹了關于Python控制臺輸出俄羅斯方塊的相關資料,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧
2021-04-04
解決django接口無法通過ip進行訪問的問題
這篇文章主要介紹了解決django接口無法通過ip進行訪問的問題，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2020-03-03
python?Tkinter實例詳解
tkinter(Tk?interface)是Python的標準GUl庫，支持跨平臺的GUl程序開發(fā)。tkinter適合小型的GUl程序編寫，也特別適合初學者學習GUl編程，這篇文章主要介紹了python?Tkinter詳解,需要的朋友可以參考下
2023-03-03
如何使用python編寫一個簡單的課時記錄系統(tǒng)
編寫一個應用系統(tǒng)需要多方面的知識和技能,下面這篇文章主要給大家介紹了關于如何使用python編寫一個簡單的課時記錄系統(tǒng)的相關資料,文中通過代碼介紹的非常詳細,對大家學習或者使用python具有一定的參考借鑒價值,需要的朋友可以參考下
2024-04-04
如何使用yolov5輸出檢測到的目標坐標信息
YOLOv5是一系列在 COCO 數(shù)據(jù)集上預訓練的對象檢測架構和模型,下面這篇文章主要給大家介紹了關于如何使用yolov5輸出檢測到的目標坐標信息的相關資料,需要的朋友可以參考下
2022-03-03
Python字符串詳細介紹
這篇文章主要介紹了Python字符串詳解,本文講解了字符串相關知識、字符串的一些特性、原始字符串、unicode字符串、字符串的常用操作方法、內(nèi)建函數(shù)列表等內(nèi)容,需要的朋友可以參考下
2015-05-05