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