PyTorch函數(shù)torch.cat與torch.stac的區(qū)別小結(jié)
更新時間：2023年09月08日 11:19:57 作者：枯木何日可逢春
Pytorch中常用的兩個拼接函數(shù)torch.cat() 和 torch.stack(),本文主要介紹了這兩個函數(shù)的用法加區(qū)別,具有一定的參考價值,感興趣的可以了解一下
一、torch.cat與torch.stack的區(qū)別
torch.cat 用于在給定的維度上連接多個張量，它將這些張量沿著指定維度堆疊在一起。
torch.stack 用于在新的維度上堆疊多個張量，它會創(chuàng)建一個新的維度，并將這些張量沿著這個新維度堆疊在一起。
二、torch.cat
Example1：
import torch tensor1 = torch.tensor([[1, 2], [3, 4]]) tensor2 = torch.tensor([[5, 6], [7, 8]]) result1 = torch.cat((tensor1, tensor2), dim=0) result2 = torch.cat((tensor1, tensor2), dim=1) print(result1.shape) print(result1) print(result2.shape) print(result2)
torch.Size([4, 2])
tensor([[1, 2],
[3, 4],
[5, 6],
[7, 8]])
torch.Size([2, 4])
tensor([[1, 2, 5, 6],
[3, 4, 7, 8]])三、torch.stack
Example1：
import torch tensor1 = torch.tensor([1, 2, 3]) tensor2 = torch.tensor([4, 5, 6]) result1 = torch.stack((tensor1, tensor2), dim=0) result2 = torch.stack((tensor1, tensor2), dim=1) print(result1.shape) print(result1) print(result2.shape) print(result2)
torch.Size([2, 3])
tensor([[1, 2, 3],
[4, 5, 6]])
torch.Size([3, 2])
tensor([[1, 4],
[2, 5],
[3, 6]])Example2：
import torch tensor1 = torch.tensor([[1, 2], [3, 4], [5, 6]]) tensor2 = torch.tensor([[7, 8], [9, 10], [11, 12]]) tensor3 = torch.tensor([[13, 14], [15, 16], [17, 18]]) result1 = torch.stack((tensor1, tensor2, tensor3), dim=0) result2 = torch.stack((tensor1, tensor2, tensor3), dim=1) print(result1.shape) print(result1) print(result2.shape) print(result2)
torch.Size([3, 3, 2])
tensor([[[ 1, 2],
[ 3, 4],
[ 5, 6]],
[[ 7, 8],
[ 9, 10],
[11, 12]],
[[13, 14],
[15, 16],
[17, 18]]])
torch.Size([3, 3, 2])
tensor([[[ 1, 2],
[ 7, 8],
[13, 14]],
[[ 3, 4],
[ 9, 10],
[15, 16]],
[[ 5, 6],
[11, 12],
[17, 18]]])到此這篇關(guān)于PyTorch函數(shù)torch.cat與torch.stac的區(qū)別小結(jié)的文章就介紹到這了,更多相關(guān)PyTorch torch.cat與torch.stac 內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
這篇文章主要介紹了用Python解數(shù)獨的方法示例，文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2019-10-10
這篇文章主要介紹了利用python模擬菜刀反彈shell繞過限制,本文給大家介紹的非常詳細，具有一定的參考借鑒價值，需要的朋友可以參考下2019-06-06
這篇文章主要介紹了Python中Django框架利用url來控制登錄的方法,實例分析了Django框架實現(xiàn)URL登陸的技巧,具有一定參考借鑒價值,需要的朋友可以參考下2015-07-07
這篇文章主要為大家詳細介紹了python版單鏈表反轉(zhuǎn)，文中示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下2022-05-05
這篇文章主要介紹了Python實現(xiàn)GUI圖片瀏覽程序,程序的實現(xiàn)需要pillow庫,pillow是 Python 的第三方圖像處理庫,需要安裝才能實用,文中通過代碼示例給大家介紹的非常詳細,需要的朋友可以參考下2023-12-12
這篇文章主要介紹了詳解將Django部署到Centos7全攻略，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2018-09-09
這篇文章主要介紹了Python保存MongoDB上的文件到本地的方法,涉及Python調(diào)用pymongo模塊的gridfs方法來操作MongoDB文件的相關(guān)技巧,需要的朋友可以參考下2016-03-03
Python中l(wèi)ist列表添加元素的3種方法總結(jié)
這篇文章主要介紹了Python中l(wèi)ist列表添加元素的3種方法總結(jié)，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2023-01-01
這篇文章主要介紹了Django單元測試中Fixtures用法，通過實例代碼給大家介紹的非常詳細，具有一定的參考借鑒價值，需要的朋友可以參考下2020-02-02
這篇文章主要介紹了詳解Python中的各種轉(zhuǎn)義符\n\r\t，文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2019-07-07