Pytorch中的dim操作介紹

本篇內(nèi)容介紹了“Pytorch 中的 dim操作介紹”的有關(guān)知識，在實際案例的操作過程中，不少人都會遇到這樣的困境，接下來就讓小編帶領(lǐng)大家學(xué)習(xí)一下如何處理這些情況吧！希望大家仔細(xì)閱讀，能夠?qū)W有所成！

成都創(chuàng)新互聯(lián)公司專注于烏爾禾網(wǎng)站建設(shè)服務(wù)及定制，我們擁有豐富的企業(yè)做網(wǎng)站經(jīng)驗。 熱誠為您提供烏爾禾營銷型網(wǎng)站建設(shè)，烏爾禾網(wǎng)站制作、烏爾禾網(wǎng)頁設(shè)計、烏爾禾網(wǎng)站官網(wǎng)定制、小程序制作服務(wù)，打造烏爾禾網(wǎng)絡(luò)公司原創(chuàng)品牌,更為您提供烏爾禾網(wǎng)站排名全網(wǎng)營銷落地服務(wù)。

Pytorch 中對 tensor 的很多操作如sum、argmax、等都可以設(shè)置dim參數(shù)用來指定操作在哪一維進(jìn)行。Pytorch 中的 dim 類似于 numpy 中的 axis，這篇文章來總結(jié)一下 Pytorch 中的 dim 操作。

dim 與方括號的關(guān)系

創(chuàng)建一個矩陣

a = torch.tensor([[1, 2], [3, 4]])
print(a)

輸出

tensor([[1, 2],
        [3, 4]])

因為a是一個矩陣，所以a的左邊有 2 個括號

括號之間是嵌套關(guān)系，代表了不同的維度。從左往右數(shù)，兩個括號代表的維度分別是 0 和 1 ，在第 0 維遍歷得到向量，在第 1 維遍歷得到標(biāo)量

同樣地，對于 3 維 tensor

b = torch.tensor([[[3, 2], [1, 4]], [[5, 6], [7, 8]]])
print(b)

輸出

tensor([[[3, 2],
         [1, 4]],

        [[5, 6],
         [7, 8]]])

則 3 個括號代表的維度從左往右分別為 0, 1, 2，在第 0 維遍歷得到矩陣，在第 1 維遍歷得到向量，在第 2 維遍歷得到標(biāo)量

更詳細(xì)一點

在指定的維度上進(jìn)行操作

在某一維度求和（或者進(jìn)行其他操作）就是對該維度中的元素進(jìn)行求和。
對于矩陣 a

a = torch.tensor([[1, 2], [3, 4]])
print(a)

輸出

tensor([[1, 2],
        [3, 4]])

求 a 在第 0 維的和，因為第 0 維代表最外邊的括號，括號中的元素為向量[1, 2]，[3, 4]，第 0 維的和就是第 0 維中的元素相加，也就是兩個向量[1, 2]，[3, 4]相加，所以結(jié)果為

s = torch.sum(a, dim=0)
print(s)

輸出

tensor([4, 6])

可以看到，a 是 2 維矩陣，而相加的結(jié)果為 1 維向量，可以使用參數(shù)keepdim=True來保證形狀不變

s = torch.sum(a, dim=0, keepdim=True)
print(s)

輸出

tensor([[4, 6]])

在 a 的第 0 維求和，就是對第 0 維中的元素（向量）進(jìn)行相加。同樣的，對 a 第 1 維求和，就是對 a 第 1 維中的元素（標(biāo)量）進(jìn)行相加，a 的第 1 維元素為標(biāo)量 1，2 和 3，4，則結(jié)果為

s = torch.sum(a, dim=1)
print(s)

輸出

tensor([3, 7])

保持維度不變

s = torch.sum(a, dim=1, keepdim=True)
print(s)

輸出

tensor([[3],
        [7]])

對 3 維 tensor 的操作也是這樣

b = torch.tensor([[[3, 2], [1, 4]], [[5, 6], [7, 8]]])
print(b)

輸出

tensor([[[3, 2],
         [1, 4]],

        [[5, 6],
         [7, 8]]])

將 b 在第 0 維相加，第 0 維為最外層括號，最外層括號中的元素為矩陣[[3, 2], [1, 4]]和[[5, 6], [7, 8]]。在第 0 維求和，就是將第 0 維中的元素（矩陣）相加

s = torch.sum(b, dim=0)
print(s)

輸出

tensor([[ 8,  8],
        [ 8, 12]])

求 b 在第 1 維的和，就是將 b 第 1 維中的元素[3, 2]和[1, 4],[5, 6]和[7, 8]相加，所以

s = torch.sum(b, dim=1)
print(s)

輸出

tensor([[ 4,  6],
        [12, 14]])

則在 b 的第 2 維求和，就是對標(biāo)量 3 和 2, 1 和 4, 5 和 6 , 7 和 8 求和

s = torch.sum(b, dim=2)
print(s)

結(jié)果為

tensor([[ 5,  5],
        [11, 15]])

除了求和，其他操作也是類似的，如求 b 在指定維度上的最大值

m = torch.max(b, dim=0)
print(m)

b 在第 0 維的最大值是第 0 維中的元素（兩個矩陣[[3, 2], [1, 4]]和[[5, 6], [7, 8]]）的最大值，取矩陣對應(yīng)位置最大值即可
結(jié)果為

torch.return_types.max(
values=tensor([[5, 6],
        [7, 8]]),
indices=tensor([[1, 1],
        [1, 1]]))

b 在第 1 維的最大值就是第 1 維元素（4 個（2對）向量）的最大值

m = torch.max(b, dim=1)
print(m)

輸出為

torch.return_types.max(
values=tensor([[3, 4],
        [7, 8]]),
indices=tensor([[0, 1],
        [1, 1]]))

b 在第 0 維的最大值就是第 0 為元素（8 個（4 對）標(biāo)量）的最大值

m = torch.max(b, dim=2)
print(m)

輸出

torch.return_types.max(
values=tensor([[3, 4],
        [6, 8]]),
indices=tensor([[0, 1],
        [1, 1]]))

總結(jié)

在 tensor 的指定維度操作就是對指定維度包含的元素進(jìn)行操作，如果想要保持結(jié)果的維度不變，設(shè)置參數(shù)keepdim=True即可。

“Pytorch 中的 dim操作介紹”的內(nèi)容就介紹到這里了，感謝大家的閱讀。如果想了解更多行業(yè)相關(guān)的知識可以關(guān)注創(chuàng)新互聯(lián)網(wǎng)站，小編將為大家輸出更多高質(zhì)量的實用文章！

當(dāng)前題目：Pytorch中的dim操作介紹
文章地址：http://muchs.cn/article4/pppoie.html

成都網(wǎng)站建設(shè)公司_創(chuàng)新互聯(lián)，為您提供網(wǎng)站營銷、做網(wǎng)站、面包屑導(dǎo)航、移動網(wǎng)站建設(shè)、App開發(fā)、外貿(mào)建站

廣告

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