利用Pytorch如何實(shí)現(xiàn)壓縮Tensor維度和擴(kuò)展Tensor維度-創(chuàng)新互聯(lián)

這篇文章將為大家詳細(xì)講解有關(guān)利用Pytorch如何實(shí)現(xiàn)壓縮Tensor維度和擴(kuò)展Tensor維度，文章內(nèi)容質(zhì)量較高，因此小編分享給大家做個參考，希望大家閱讀完這篇文章后對相關(guān)知識有一定的了解。

專注于為中小企業(yè)提供成都網(wǎng)站制作、網(wǎng)站建設(shè)服務(wù),電腦端+手機(jī)端+微信端的三站合一,更高效的管理,為中小企業(yè)朔州免費(fèi)做網(wǎng)站提供優(yōu)質(zhì)的服務(wù)。我們立足成都，凝聚了一批互聯(lián)網(wǎng)行業(yè)人才，有力地推動了超過千家企業(yè)的穩(wěn)健成長，幫助中小企業(yè)通過網(wǎng)站建設(shè)實(shí)現(xiàn)規(guī)模擴(kuò)充和轉(zhuǎn)變。

1. 擴(kuò)展Tensor維度

　　相信剛接觸Pytorch的寶寶們，會遇到這樣一個問題，輸入的數(shù)據(jù)維度和實(shí)驗(yàn)需要維度不一致，輸入的可能是2維數(shù)據(jù)或3維數(shù)據(jù)，實(shí)驗(yàn)需要用到3維或4維數(shù)據(jù)，那么我們需要擴(kuò)展這個維度。其實(shí)特別簡單，只要對數(shù)據(jù)加一個擴(kuò)展維度方法就可以了。

1.1torch.unsqueeze(self: Tensor, dim: _int)

　　torch.unsqueeze(self: Tensor, dim: _int)

　　參數(shù)說明：self:輸入的tensor數(shù)據(jù)，dim：要對哪個維度擴(kuò)展就輸入那個維度的整數(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)

本文題目：利用Pytorch如何實(shí)現(xiàn)壓縮Tensor維度和擴(kuò)展Tensor維度-創(chuàng)新互聯(lián)
本文URL：http://bm7419.com/article38/dgoipp.html

成都網(wǎng)站建設(shè)公司_創(chuàng)新互聯(lián)，為您提供Google、面包屑導(dǎo)航、外貿(mào)網(wǎng)站建設(shè)、關(guān)鍵詞優(yōu)化、標(biāo)簽優(yōu)化、微信公眾號

廣告

聲明：本網(wǎng)站發(fā)布的內(nèi)容（圖片、視頻和文字）以用戶投稿、用戶轉(zhuǎn)載內(nèi)容為主，如果涉及侵權(quán)請盡快告知，我們將會在第一時間刪除。文章觀點(diǎn)不代表本網(wǎng)站立場，如需處理請聯(lián)系客服。電話：028-86922220；郵箱：631063699@qq.com。內(nèi)容未經(jīng)允許不得轉(zhuǎn)載，或轉(zhuǎn)載時需注明來源： 創(chuàng)新互聯(lián)