pytorch torch.expand和torch.repeat的區(qū)別詳解
1.torch.expand
函數(shù)返回張量在某一個(gè)維度擴(kuò)展之后的張量，就是將張量廣播到新形狀。函數(shù)對(duì)返回的張量不會(huì)分配新內(nèi)存，即在原始張量上返回只讀視圖，返回的張量?jī)?nèi)存是不連續(xù)的。類似于numpy中的broadcast_to函數(shù)的作用。如果希望張量?jī)?nèi)存連續(xù)，可以調(diào)用contiguous函數(shù)。
例子：
import torch x = torch.tensor([1, 2, 3, 4]) xnew = x.expand(2, 4) print(xnew)
輸出：
tensor([[1, 2, 3, 4],
[1, 2, 3, 4]])
2.torch.repeat
torch.repeat用法類似np.tile，就是將原矩陣橫向、縱向地復(fù)制。與torch.expand不同的是torch.repeat返回的張量在內(nèi)存中是連續(xù)的。
例子1：
將張量橫向的復(fù)制
import torch x = torch.tensor([1, 2, 3]) xnew = x.repeat(1,3) print(xnew)
輸出：
tensor([[1, 2, 3, 1, 2, 3, 1, 2, 3]])
例子2：
將張量縱向的復(fù)制
import torch x = torch.tensor([1, 2, 3]) xnew = x.repeat(3,1) print(xnew)
輸出：
tensor([[1, 2, 3],
[1, 2, 3],
[1, 2, 3]])
以上就是本文的全部?jī)?nèi)容，希望對(duì)大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。
相關(guān)文章
這篇文章主要介紹了python opencv之SURF算法示例,小編覺得挺不錯(cuò)的，現(xiàn)在分享給大家，也給大家做個(gè)參考。一起跟隨小編過來(lái)看看吧2018-02-02
python-itchat 獲取微信群用戶信息的實(shí)例
今天小編就為大家分享一篇python-itchat 獲取微信群用戶信息的實(shí)例，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來(lái)看看吧2019-02-02
Python卷積神經(jīng)網(wǎng)絡(luò)圖片分類框架詳解分析
在機(jī)器視覺領(lǐng)域中,卷積神經(jīng)網(wǎng)絡(luò)算法作為一種新興算法出現(xiàn),在圖像識(shí)別領(lǐng)域中,卷積神經(jīng)網(wǎng)絡(luò)能夠較好的實(shí)現(xiàn)圖像的分類效果,而且其位移和形變具有較高的容忍能力2021-11-11
基于Python實(shí)現(xiàn)一個(gè)簡(jiǎn)易的數(shù)據(jù)管理系統(tǒng)
為了方便的實(shí)現(xiàn)記錄數(shù)據(jù)、修改數(shù)據(jù)沒有精力去做一個(gè)完整的系統(tǒng)去管理數(shù)據(jù)。因此，在python的控制臺(tái)直接實(shí)現(xiàn)一個(gè)簡(jiǎn)易的數(shù)據(jù)管理系統(tǒng)，包括數(shù)據(jù)的增刪改查等等。感興趣的可以跟隨小編一起學(xué)習(xí)一下2021-12-12
Python異步在非阻塞子進(jìn)程中運(yùn)行命令詳解
這篇文章主要為大家介紹了Python異步在非阻塞子進(jìn)程中運(yùn)行命令詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪2023-03-03
Python3利用openpyxl讀寫Excel文件的方法實(shí)例
這篇文章主要給大家介紹了關(guān)于Python3利用openpyxl讀寫Excel文件的相關(guān)資料，文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2021-02-02
Python編程實(shí)現(xiàn)雙鏈表，棧，隊(duì)列及二叉樹的方法示例
這篇文章主要介紹了Python編程實(shí)現(xiàn)雙鏈表，棧，隊(duì)列及二叉樹的方法,結(jié)合具體實(shí)例形式分析了Python簡(jiǎn)單實(shí)現(xiàn)數(shù)據(jù)結(jié)構(gòu)中雙鏈表，棧，隊(duì)列及二叉樹相關(guān)操作技巧,需要的朋友可以參考下2017-11-11
這篇文章主要介紹了python實(shí)操案例練習(xí)，本文給大家分享的案例中主要任務(wù)有編寫程序?qū)崿F(xiàn)樂手彈奏樂器、使用面向?qū)ο笤O(shè)計(jì)自定義類，描述出租車和家用轎車的信息，需要的小伙伴可以參考一下2022-02-02
這篇文章主要介紹了Python面向?qū)ο蠓庋b操作,結(jié)合具體案例形式詳細(xì)分析了Python面向?qū)ο蟮姆庋b、擴(kuò)展等操作技巧與相關(guān)注意事項(xiàng),需要的朋友可以參考下2019-12-12
基于Python實(shí)現(xiàn)計(jì)算納什均衡的示例詳解
納什均衡是一種博弈論中的概念，它描述了一種平衡狀態(tài)，其中每個(gè)參與者都不能通過獨(dú)立改變其決策來(lái)提高自己的利益。本文就來(lái)用Python中的Nashpy和PuLP實(shí)現(xiàn)計(jì)算納什均衡，感興趣的可以了解一下2023-02-02