解決ImportError:cannot import name ‘Flatten‘ from ‘torch.nn‘問(wèn)題
更新時(shí)間：2023年06月15日 09:09:58 作者：全幼兒園最聰明
這篇文章主要介紹了解決ImportError:cannot import name ‘Flatten‘ from ‘torch.nn‘問(wèn)題，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教
問(wèn)題描述
在代碼中從torch.nn引入Flatten報(bào)錯(cuò)：
解決方案
進(jìn)入python交互式環(huán)境
輸入以下代碼，查看pytorch版本：
import torch print(torch.__version__)
可以看到pytorch版本是1.1.0
版本太低不包含Flatten屬性
因此，一個(gè)辦法就是升級(jí)pytorch版本，但我們知道下載安裝pytorch并不容易，若沒(méi)有鏡像下載緩慢且需要考慮顯卡驅(qū)動(dòng)、conda、torch三者的版本關(guān)系，升級(jí)可能導(dǎo)致版本不匹配conda無(wú)法使用gpu等問(wèn)題。
另一個(gè)簡(jiǎn)單的辦法是自己寫(xiě)Flatten類，這里我需要的功能是僅是扁平化，
參考代碼如下：
class Flatten(nn.Module):
def forward(self, input):
return input.view(input.size(0), -1)引入后重新運(yùn)行代碼，沒(méi)有報(bào)錯(cuò)
總結(jié)
以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
您可能感興趣的文章:
- 關(guān)于Python ImportError: No module named 通用解決方法
- python之ImportError:模塊引入異常問(wèn)題
- python解決報(bào)錯(cuò)ImportError: Bad git executable.問(wèn)題
- 安裝pyhttpx解決ImportError: DLL load failed錯(cuò)誤
- 解決Numpy報(bào)錯(cuò):ImportError: numpy.core.multiarray failed
- 關(guān)于sklearn包導(dǎo)入錯(cuò)誤:ImportError:?cannot?import?name Type解決方案
- Python報(bào)錯(cuò)ImportError:?IProgress?not?found.?Please?update?jupyter?and?ipywidgets解決
- Python報(bào)錯(cuò)ImportError: No module named ‘missing_module‘的解決方法
- Python解決“ImportError:?Couldn‘t?import?Django”問(wèn)題全攻略
相關(guān)文章
jupyter notebook 多環(huán)境conda kernel配置方式
這篇文章主要介紹了jupyter notebook 多環(huán)境conda kernel配置方式，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2020-04-04
基于Python?OpenCV和?dlib實(shí)現(xiàn)眨眼檢測(cè)
這篇文章主要介紹了基于Python?OPenCV及dlib實(shí)現(xiàn)檢測(cè)視頻流中的眨眼次數(shù)。文中的代碼對(duì)我們的學(xué)習(xí)和工作有一定價(jià)值，感興趣的同學(xué)可以參考一下2021-12-12
用Python實(shí)現(xiàn)一個(gè)簡(jiǎn)單的能夠發(fā)送帶附件的郵件程序的教程
這篇文章主要介紹了用Python實(shí)現(xiàn)一個(gè)簡(jiǎn)單的能夠發(fā)送帶附件的郵件程序的教程,用MIMEApplication模塊來(lái)發(fā)送各種類型的文件,需要的朋友可以參考下2015-04-04
Keras 數(shù)據(jù)增強(qiáng)ImageDataGenerator多輸入多輸出實(shí)例
這篇文章主要介紹了Keras 數(shù)據(jù)增強(qiáng)ImageDataGenerator多輸入多輸出實(shí)例，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2020-07-07
為Python的Tornado框架配置使用Jinja2模板引擎的方法
Jinja2是人氣Web框架Flask中的內(nèi)置模板引擎,而且與Django的模板引擎比較類似,這里我們就來(lái)看一下為Python的Tornado框架配置使用Jinja2模板引擎的方法2016-06-06
Selenium向iframe富文本框輸入內(nèi)容過(guò)程圖解
這篇文章主要介紹了Selenium向iframe富文本框輸入內(nèi)容過(guò)程圖解,文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2020-04-04
這篇文章主要介紹了pyspark操作MongoDB的方法步驟，小編覺(jué)得挺不錯(cuò)的，現(xiàn)在分享給大家，也給大家做個(gè)參考。一起跟隨小編過(guò)來(lái)看看吧2019-01-01
這段文章詳細(xì)講解了Python元組的用法,從基礎(chǔ)到高級(jí)技巧,涵蓋創(chuàng)建、解包、嵌套、與列表字典集合的轉(zhuǎn)換等,還強(qiáng)調(diào)了元組在表示固定數(shù)據(jù)、防止誤改、作為字典鍵等方面的優(yōu)勢(shì),適合Python新手理解和掌握,需要的朋友可以參考下2026-05-05
本文給大家介紹Python中l(wèi)ist.sort()和sorted()的使用區(qū)別,詳細(xì)介紹其參數(shù)功能及Timsort排序算法特性,涵蓋自適應(yīng)性、穩(wěn)定性、時(shí)間復(fù)雜度,并提及該算法在Java、JavaScript等多語(yǔ)言中的應(yīng)用,感興趣的朋友跟隨小編一起看看吧2025-08-08
PyCharm Terminal終端命令行Shell設(shè)置方式
這篇文章主要介紹了PyCharm Terminal終端命令行Shell設(shè)置方式，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教2023-01-01