pytorch?tensor合并與分割方式

更新時間：2024年02月20日 09:09:10 作者：wyw0000

這篇文章主要介紹了pytorch?tensor合并與分割方式,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教

1. cat

torch.cat(tensors, dim=0, *, out=None) → Tensor

在指定維度上，連接給定tensor序列或empty，除連接的dimension外，所有得的ensor必須有相同的shape

參數(shù)：

tensors-具有相同類型的tensor序列，非empty tensor必須具有相同的shape，連接的dimension除外
dim-指定的連接的維度

輸出：

連接后的tensor

上圖分別是在列和行兩個維度連接后的結(jié)果

2. stack

創(chuàng)建新維度來連接張量序列

torch.stack(tensors, dim=0, *, out=None) → Tensor

參數(shù)：

tensors-張量序列，具有相同的size
dim-插入的新維度，必須介于0和連接的tensor的維度之間

輸出：

連接后的tensor

注意：cat和stack的區(qū)別

stack連接的tensor必須具有相同的size，否則報錯，cat是除連接的維度外，其他維度shape必須相同

如下示例：

3. split

把一個tensor切分成塊，每個塊是原tensor的一部分

torch.split(tensor, split_size_or_sections, dim=0)

參數(shù)：

tensor-用來切分的tensor
split_size_or_sections (int) or (list(int)) -單個塊的size后者是每個塊size的list
dim (int) – 以tensor的哪個維度進行切分

輸出：

Tuple[Tensor, …]

示例：

4. chunk

強制將一個tensor切分成指定數(shù)量的塊，每個塊是原tensor的一部分

torch.chunk(input, chunks, dim=0) → List of Tensors

參數(shù)：

input (Tensor) – 輸入切分的tensor
chunks (int) – 切分塊的數(shù)量
dim (int) – 以tensor的哪個維度進行切分

輸出：

切分后的list

示例：

注意：split與chunk的區(qū)別

區(qū)別主要是第二個參數(shù)，split第二個參數(shù)切分塊的size，而chunk是切分塊的數(shù)量

總結(jié)

以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家。

您可能感興趣的文章:

Python實現(xiàn)自動添加腳本頭信息的示例代碼
這篇文章給大家介紹的一段腳本是自動添加注釋信息的腳本，添加的信息包括腳本名稱、作者和時間等之類的，對于團隊形成統(tǒng)一的編碼規(guī)則很有幫助。有需要的可以參考借鑒。
2016-09-09
python自動化測試之setUp與tearDown實例
這篇文章主要介紹了python自動化測試之setUp與tearDown實例,其中setUp()方法中進行測試前的初始化工作,并在tearDown()方法中執(zhí)行測試后的清除工作,setUp()和tearDown()都是TestCase類中定義的方法,需要的朋友可以參考下
2014-09-09
pandas to_excel 添加顏色操作
這篇文章主要介紹了pandas to_excel 添加顏色操作，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2020-07-07
Django ORM查詢操作方式
Django提供了一套非常方便的類似SqlAlchemy ORM的通過對象調(diào)用的方式操作數(shù)據(jù)庫表的ORM框架,,本文給大家詳細介紹Django ORM查詢操作方式,感興趣的朋友一起看看吧
2023-10-10
Python?中?Pandas?文件操作和讀取?CSV?參數(shù)詳解
CSV?又稱逗號分隔值文件，是一種簡單的文件格式，以特定的結(jié)構(gòu)來排列表格數(shù)據(jù)，這篇文章主要介紹了Python?之?Pandas?文件操作和讀取?CSV?參數(shù)詳解,需要的朋友可以參考下
2023-03-03
python記錄程序運行時間的三種方法
這篇文章主要介紹了python記錄程序運行時間的三種方法的相關資料,需要的朋友可以參考下
2017-07-07
python列表和字符串的三種逆序遍歷操作
這篇文章主要介紹了python列表和字符串的三種逆序遍歷操作，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教
2021-06-06
python與js進行MD5取hash有什么不同
這篇文章主要講解得內(nèi)容是python與js進行MD5取hash有什么不同，我們在做前端做滲透測試時會遇到一些關鍵字進行了加密得情況，而且python和js對json進行md5取hash,MD5結(jié)果值還不一致，下面我們就爛看看到底是哪里不同吧，需要的朋友可以參考一下
2022-02-02
python 實現(xiàn)保存最新的三份文件,其余的都刪掉
今天小編就為大家分享一篇python 實現(xiàn)保存最新的三份文件,其余的都刪掉，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2019-12-12
python判斷圖片相似度找出相似的圖像問題
這篇文章主要介紹了python判斷圖片相似度找出相似的圖像問題,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教
2026-03-03