pytorch中的 .view()函數(shù)的用法介紹
一、普通用法 (手動調(diào)整size)
view()相當(dāng)于reshape、resize,重新調(diào)整Tensor的形狀。
import torch a1 = torch.arange(0,16) print(a1) # tensor([ 0, ?1, ?2, ?3, ?4, ?5, ?6, ?7, ?8, ?9, 10, 11, 12, 13, 14, 15])
a2 = a1.view(8, 2) a3 = a1.view(2, 8) a4 = a1.view(4, 4) print(a2) #tensor([[ 0, ?1], # ? ? ? ?[ 2, ?3], # ? ? ? ?[ 4, ?5], # ? ? ? ?[ 6, ?7], # ? ? ? ?[ 8, ?9], # ? ? ? ?[10, 11], # ? ? ? ?[12, 13], # ? ? ? ?[14, 15]]) print(a3) #tensor([[ 0, ?1, ?2, ?3, ?4, ?5, ?6, ?7], # ? ? ? ?[ 8, ?9, 10, 11, 12, 13, 14, 15]]) print(a4) #tensor([[ 0, ?1, ?2, ?3], # ? ? ? ?[ 4, ?5, ?6, ?7], # ? ? ? ?[ 8, ?9, 10, 11], # ? ? ? ?[12, 13, 14, 15]])
二、特殊用法:參數(shù)-1 (自動調(diào)整size)
view中一個(gè)參數(shù)定為-1,代表自動調(diào)整這個(gè)維度上的元素個(gè)數(shù),以保證元素的總數(shù)不變。
v1 = torch.arange(0,16) print(v1) # tensor([ 0, ?1, ?2, ?3, ?4, ?5, ?6, ?7, ?8, ?9, 10, 11, 12, 13, 14, 15]) v2 = v1.view(-1, 16) v2 # tensor([[ 0, ?1, ?2, ?3, ?4, ?5, ?6, ?7, ?8, ?9, 10, 11, 12, 13, 14, 15]]) v2 = v1.view(-1, 8) v2 # tensor([[ 0, ?1, ?2, ?3, ?4, ?5, ?6, ?7], # ? ? ? ? [ 8, ?9, 10, 11, 12, 13, 14, 15]]) v2 = v1.view(-1, 4) v2 #tensor([[ 0, ?1, ?2, ?3], # ? ? ? ?[ 4, ?5, ?6, ?7], # ? ? ? ?[ 8, ?9, 10, 11], # ? ? ? ?[12, 13, 14, 15]]) v2 = v1.view(-1, 2) v2 #tensor([[ 0, ?1], # ? ? ? ?[ 2, ?3], # ? ? ? ?[ 4, ?5], # ? ? ? ?[ 6, ?7], # ? ? ? ?[ 8, ?9], # ? ? ? ?[10, 11], # ? ? ? ?[12, 13], # ? ? ? ?[14, 15]])
v3 = v1.view(4*4, -1) v3 # tensor([[ 0], # ? ? ? ? [ 1], # ? ? ? ? [ 2], # ? ? ? ? [ 3], # ? ? ? ? [ 4], # ? ? ? ? [ 5], # ? ? ? ? [ 6], # ? ? ? ? [ 7], # ? ? ? ? [ 8], # ? ? ? ? [ 9], # ? ? ? ? [10], # ? ? ? ? [11], # ? ? ? ? [12], # ? ? ? ? [13], # ? ? ? ? [14], # ? ? ? ? [15]])
到此這篇關(guān)于pytorch中的 .view()函數(shù)的用法介紹的文章就介紹到這了,更多相關(guān)pytorch .view()函數(shù)內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
相關(guān)文章
python中如何正確使用正則表達(dá)式的詳細(xì)模式(Verbose mode expression)
許多程序設(shè)計(jì)語言都支持利用正則表達(dá)式進(jìn)行字符串操作,python自然也不例外,下面這篇文章主要給大家介紹了關(guān)于在python中如何正確使用正則表達(dá)式的詳細(xì)模式(Verbose mode expression)的相關(guān)資料,需要的朋友可以參考借鑒,下面來一起看看吧。2017-11-11Python中的數(shù)據(jù)對象持久化存儲模塊pickle的使用示例
這篇文章主要介紹了Python中的數(shù)據(jù)對象持久化存儲模塊pickle的使用示例,重點(diǎn)講解了pickle中模塊中對象持久化和文件讀取的相關(guān)方法,需要的朋友可以參考下2016-03-03python polars數(shù)據(jù)科學(xué)庫對比Pandas優(yōu)勢分析
這篇文章主要為大家介紹了python polars數(shù)據(jù)科學(xué)庫對比Pandas優(yōu)勢分析,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2024-01-01Python序列對象與String類型內(nèi)置方法詳解
這篇文章主要介紹了Python序列對象與String類型內(nèi)置方法,結(jié)合實(shí)例形式分析了Python序列對象與String類型各種常見內(nèi)置方法相關(guān)使用技巧及操作注意事項(xiàng),需要的朋友可以參考下2019-10-10如何通過雪花算法用Python實(shí)現(xiàn)一個(gè)簡單的發(fā)號器
這篇文章主要介紹了如何通過雪花算法用Python實(shí)現(xiàn)一個(gè)簡單的發(fā)號器,文中通過示例代碼介紹的非常詳細(xì),對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2019-07-07利用Python實(shí)現(xiàn)簡單的Excel統(tǒng)計(jì)函數(shù)
這篇文章主要介紹了利用Python實(shí)現(xiàn)簡單的Excel統(tǒng)計(jì)函數(shù),文章圍繞主題展開詳細(xì)的內(nèi)容介紹,具有一定的參考價(jià)值,需要的小伙伴可以參考一下2022-07-07利用Pycharm斷點(diǎn)調(diào)試Python程序的方法
今天小編就為大家分享一篇利用Pycharm斷點(diǎn)調(diào)試Python程序的方法,具有很好的參考價(jià)值,希望對大家有所幫助。一起跟隨小編過來看看吧2018-11-11