解決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并不容易，若沒有鏡像下載緩慢且需要考慮顯卡驅(qū)動、conda、torch三者的版本關(guān)系，升級可能導(dǎo)致版本不匹配conda無法使用gpu等問題。

另一個簡單的辦法是自己寫Flatten類，這里我需要的功能是僅是扁平化，

參考代碼如下：

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

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

總結(jié)

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

您可能感興趣的文章:

相關(guān)文章

jupyter notebook 多環(huán)境conda kernel配置方式
這篇文章主要介紹了jupyter notebook 多環(huán)境conda kernel配置方式，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2020-04-04
基于Python?OpenCV和?dlib實現(xiàn)眨眼檢測
這篇文章主要介紹了基于Python?OPenCV及dlib實現(xiàn)檢測視頻流中的眨眼次數(shù)。文中的代碼對我們的學(xué)習(xí)和工作有一定價值，感興趣的同學(xué)可以參考一下
2021-12-12
用Python實現(xiàn)一個簡單的能夠發(fā)送帶附件的郵件程序的教程
這篇文章主要介紹了用Python實現(xiàn)一個簡單的能夠發(fā)送帶附件的郵件程序的教程,用MIMEApplication模塊來發(fā)送各種類型的文件,需要的朋友可以參考下
2015-04-04
Keras 數(shù)據(jù)增強ImageDataGenerator多輸入多輸出實例
這篇文章主要介紹了Keras 數(shù)據(jù)增強ImageDataGenerator多輸入多輸出實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2020-07-07
為Python的Tornado框架配置使用Jinja2模板引擎的方法
Jinja2是人氣Web框架Flask中的內(nèi)置模板引擎,而且與Django的模板引擎比較類似,這里我們就來看一下為Python的Tornado框架配置使用Jinja2模板引擎的方法
2016-06-06
Selenium向iframe富文本框輸入內(nèi)容過程圖解
這篇文章主要介紹了Selenium向iframe富文本框輸入內(nèi)容過程圖解,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下
2020-04-04
pyspark操作MongoDB的方法步驟
這篇文章主要介紹了pyspark操作MongoDB的方法步驟，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧
2019-01-01
Python元組(Tuple)全部高級用法大全
這段文章詳細(xì)講解了Python元組的用法,從基礎(chǔ)到高級技巧,涵蓋創(chuàng)建、解包、嵌套、與列表字典集合的轉(zhuǎn)換等,還強調(diào)了元組在表示固定數(shù)據(jù)、防止誤改、作為字典鍵等方面的優(yōu)勢,適合Python新手理解和掌握,需要的朋友可以參考下
2026-05-05
Python中的sort()和sorted()用法示例解析
本文給大家介紹Python中l(wèi)ist.sort()和sorted()的使用區(qū)別,詳細(xì)介紹其參數(shù)功能及Timsort排序算法特性,涵蓋自適應(yīng)性、穩(wěn)定性、時間復(fù)雜度,并提及該算法在Java、JavaScript等多語言中的應(yīng)用,感興趣的朋友跟隨小編一起看看吧
2025-08-08
PyCharm Terminal終端命令行Shell設(shè)置方式
這篇文章主要介紹了PyCharm Terminal終端命令行Shell設(shè)置方式，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教
2023-01-01