Pytorch 擴(kuò)展Tensor維度、壓縮Tensor維度的方法
1. 擴(kuò)展Tensor維度
相信剛接觸Pytorch的寶寶們，會(huì)遇到這樣一個(gè)問(wèn)題，輸入的數(shù)據(jù)維度和實(shí)驗(yàn)需要維度不一致，輸入的可能是2維數(shù)據(jù)或3維數(shù)據(jù)，實(shí)驗(yàn)需要用到3維或4維數(shù)據(jù)，那么我們需要擴(kuò)展這個(gè)維度。其實(shí)特別簡(jiǎn)單，只要對(duì)數(shù)據(jù)加一個(gè)擴(kuò)展維度方法就可以了。
1.1torch.unsqueeze(self: Tensor, dim: _int)
torch.unsqueeze(self: Tensor, dim: _int)
參數(shù)說(shuō)明：self:輸入的tensor數(shù)據(jù)，dim：要對(duì)哪個(gè)維度擴(kuò)展就輸入那個(gè)維度的整數(shù)，可以輸入0，1，2……
1.2Code
第一種方式，輸入數(shù)據(jù)后直接加unsqueeze()
擴(kuò)展第一維和第二維為1
import torch
def reset_unsqueeze1():
data = torch.rand([3, 3])
data1 = data.unsqueeze(dim=0).unsqueeze(dim=1)
print("data_size: ", data.shape)
print("data: ", data)
print("data1_size: ", data1.shape)
print("data1: ", data1)
結(jié)果顯示
第二種方式，用torch.unsqueeze()
import torch
def reset_unsqueeze2():
data = torch.rand([3, 3])
data1 = torch.unsqueeze(data, dim=0)
print("data_size: ", data.shape)
print("data: ", data)
print("data1_size: ", data1.shape)
print("data1: ", data1)
結(jié)果顯示
2. 壓縮Tensor維度
2.1torch.squeeze(self: Tensor, dim: _int)
這個(gè)方法剛好和torch.unsqueeze()方法效果相反，壓縮Tensor維度。
2.2 Code
第一種方式，輸入數(shù)據(jù)后直接加squeeze()
import torch
def reset_squeeze1():
data = torch.rand([1, 1, 3, 3])
data1 = data.squeeze(dim=0).squeeze(dim=1)
print("data_size: ", data.shape)
print("data: ", data)
print("data1_size: ", data1.shape)
print("data1: ", data1)
結(jié)果顯示
第二種方式，用torch.squeeze()
import torch
def reset_squeeze2():
data = torch.rand([1, 1, 3, 3])
data1 = torch.squeeze(data, dim=0)
print("data_size: ", data.shape)
print("data: ", data)
print("data1_size: ", data1.shape)
print("data1: ", data1)
結(jié)果顯示
到此這篇關(guān)于Pytorch 擴(kuò)展Tensor維度、壓縮Tensor維度的方法的文章就介紹到這了,更多相關(guān)Pytorch 擴(kuò)展Tensor維度、壓縮Tensor維度內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
這篇文章主要為大家介紹了Python jiaba庫(kù)的使用，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下，希望能夠給你帶來(lái)幫助，希望能夠給你帶來(lái)幫助2021-11-11
caffe的python接口之手寫(xiě)數(shù)字識(shí)別mnist實(shí)例
這篇文章主要為大家介紹了caffe的python接口之手寫(xiě)數(shù)字識(shí)別mnist實(shí)例詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪2022-06-06
Python上傳package到Pypi(代碼簡(jiǎn)單)
這篇文章主要介紹了Python上傳package到Pypi(代碼簡(jiǎn)單)的相關(guān)資料,需要的朋友可以參考下2016-02-02
Python 列表排序方法reverse、sort、sorted詳解
本文給大家介紹的是Python中列表排序方法中的reverse、sort、sorted操作方法，以及他們直接的區(qū)別介紹，有需要的小伙伴可以參考下。2016-01-01
這篇文章主要介紹了Python讀取CSV文件的四種方法， 使用 csv 模塊，使用 numpy 庫(kù)，使用pandas庫(kù)，使用標(biāo)準(zhǔn)庫(kù)中的 csv 模塊這四種方法，文章通過(guò)代碼示例介紹的非常詳細(xì),需要的朋友可以參考下2023-07-07
這篇文章主要介紹了Python 類(lèi)的特殊成員介紹,本文給大家介紹的非常詳細(xì)，具有一定的參考借鑒價(jià)值，需要的朋友可以參考下2018-06-06
這篇文章主要介紹了python SocketServer源碼深入解讀,文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2019-09-09
Python 利用4行代碼實(shí)現(xiàn)圖片灰度化的項(xiàng)目實(shí)踐
灰度處理是將彩色圖像轉(zhuǎn)換為灰度圖像的過(guò)程,即每個(gè)像素的顏色由紅、綠、藍(lán)三個(gè)通道的值組成,轉(zhuǎn)換為一個(gè)單一的灰度值,本文主要介紹了Python 利用4行代碼實(shí)現(xiàn)圖片灰度化的項(xiàng)目實(shí)踐,感興趣的可以了解一下2024-04-04
在這段時(shí)間參加了一個(gè)競(jìng)賽，寫(xiě)下了這個(gè)代碼，但是總感覺(jué)有一些地方是不完善！這是一個(gè)關(guān)于使用opencv庫(kù)判斷顏色小球形狀及位置的功能實(shí)現(xiàn)，其中也參考了一些前輩的代碼，希望能對(duì)迷茫中的小伙幫有所幫助2022-11-11
pycharm配置當(dāng)鼠標(biāo)懸停時(shí)快速提示方法參數(shù)
這篇文章主要介紹了pycharm中配置當(dāng)鼠標(biāo)懸停時(shí)快速提示方法參數(shù),本文圖文并茂給大家介紹的非常詳細(xì)，具有一定的參考借鑒價(jià)值，需要的朋友可以參考下2019-07-07