pytorch torch.expand和torch.repeat的區(qū)別詳解
更新時間：2019年11月05日 11:48:25 作者：MY Qi
這篇文章主要介紹了pytorch torch.expand和torch.repeat的區(qū)別詳解，文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧
1.torch.expand
函數(shù)返回張量在某一個維度擴展之后的張量，就是將張量廣播到新形狀。函數(shù)對返回的張量不會分配新內(nèi)存，即在原始張量上返回只讀視圖，返回的張量內(nèi)存是不連續(xù)的。類似于numpy中的broadcast_to函數(shù)的作用。如果希望張量內(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]])
以上就是本文的全部內(nèi)容，希望對大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。
相關(guān)文章
使用python讀寫txt和json(jsonl)大文件的方法步驟
在Python中讀取txt和json(jsonl)大文件并保存到字典是一項非常常見的操作,這篇文章主要給大家介紹了關(guān)于使用python讀寫txt和json(jsonl)大文件的方法步驟,需要的朋友可以參考下2023-12-12
在本篇內(nèi)容中我們給大家整理了關(guān)于python安裝Scrapy框架的圖文詳細步驟，需要的朋友們跟著學(xué)習(xí)下。2019-05-05
Python參數(shù)解析器configparser簡介
configparser是python自帶的配置參數(shù)解析器，可以用于解析.config文件中的配置參數(shù)，ini文件中由sections(節(jié)點)-key-value組成，這篇文章主要介紹了Python參數(shù)解析器configparser,需要的朋友可以參考下2022-12-12
這篇文章主要介紹了Python編程實現(xiàn)的圖片識別功能,涉及Python PIL模塊的安裝與使用技巧,需要的朋友可以參考下2017-08-08
在大家的日常python程序的編寫過程中，都會有自己解決某個問題的解決辦法，或者是在程序的調(diào)試過程中，用來幫助調(diào)試的程序公式，本文總結(jié)了22個萬用公式，感興趣的可以了解一下2021-07-07
舉例詳解Python中的split()函數(shù)的使用方法
這篇文章主要介紹了舉例詳解Python中的split()函數(shù)的使用方法,split()函數(shù)的使用是Python學(xué)習(xí)當(dāng)中的基礎(chǔ)知識,通常用于將字符串切片并轉(zhuǎn)換為列表,需要的朋友可以參考下2015-04-04
這篇文章主要為大家詳細介紹了常見編碼和解碼的基本原理,例如ASCII,GBK以及Python中的默認編碼等,感興趣的小伙伴可以跟隨小編一了解一下2025-02-02
Python判斷字符串是否包含特定子字符串的多種方法(7種方法)
我們經(jīng)常會遇這樣一個需求判斷字符串中是否包含某個關(guān)鍵詞，也就是特定的子字符串，接下來通過本文給大家分享Python判斷字符串是否包含特定子字符串的多種方法(7種方法),需要的朋友可以參考下2023-03-03
這篇文章主要為大家詳細介紹了點球小游戲python腳本，具有一定的參考價值，感興趣的小伙伴們可以參考一下2018-05-05
pyqt彈出新對話框,以及關(guān)閉對話框獲取數(shù)據(jù)的實例
今天小編就為大家分享一篇pyqt彈出新對話框,以及關(guān)閉對話框獲取數(shù)據(jù)的實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-06-06