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