pytorch中[..., 0]的用法說明
更新時間：2021年05月20日 10:30:44 作者：Qian._.
這篇文章主要介紹了pytorch中[..., 0]的用法說明，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教
在看程序的時候看到了x[…, 0]的語句不是很理解，后來自己做實驗略微了解，以此記錄方便自己查看。
b=torch.Tensor([[[[10,2],[4,5],[7,8]],[[1,2],[4,5],[7,8]]]]) print(b.size()) (1, 2, 3, 2) print(b[…,0]) tensor([[[10., 4., 7.], [ 1., 4., 7.]]]) print(b[…,0].size()) (1, 2, 3) print(b[…,2]) Traceback (most recent call last): File “”, line 1, in IndexError: index 2 is out of bounds for dimension 3 with size 2
print(b[0,…]) tensor([[[10., 2.], [ 4., 5.], [ 7., 8.]], [[ 1., 2.], [ 4., 5.], [ 7., 8.]]]) print(b[0,…].size()) (2, 3, 2)
print(b[0,…,0].size()) (2, 3) print(b[0,…,0]) tensor([[10., 4., 7.], [ 1., 4., 7.]])
[…, 0]表示抽取tensor b的第4根軸上的第一列數(shù)字組成tensor，[0, …]表示抽取tensor b的第一根軸上的第一列數(shù)字組成tensor，[0, …, 0]表示抽取b的第一根和第四根軸上的第一列數(shù)字組成tensor。
還發(fā)現(xiàn)一個現(xiàn)象
print(b[…,0:]) tensor([[[[10., 2.], [ 4., 5.], [ 7., 8.]], [[ 1., 2.], [ 4., 5.], [ 7., 8.]]]])
print(b[…,1:]) tensor([[[[2.], [5.], [8.]], [[2.], [5.], [8.]]]])
print(b[…,2:]) tensor([], size=(1, 2, 3, 0))
補充：PyTorch中[..., 0]的使用案例
1. 示例1
import torch a = torch.rand((17, 24, 8)) b = a[..., 0] c = a[:, :, 0] print(b == c)
結(jié)果為True
2. 示例2
import torch a = torch.rand((64, 17, 24, 8)) b = a[..., 0] c = a[:, :, :, 0] print(b == c)
結(jié)果為True
3. 結(jié)論
可以看出[…, 0]相當(dāng)于[:, :, … :, 0]
以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
Python實現(xiàn)判斷一個字符串是否包含子串的方法總結(jié)
這篇文章主要介紹了Python實現(xiàn)判斷一個字符串是否包含子串的方法,結(jié)合實例形式總結(jié)分析了四種比較常用的字符串子串判定方法,需要的朋友可以參考下2017-11-11
在本篇文章里小編給大家分享的是關(guān)于python中subprocess實例用法及知識點詳解內(nèi)容，有需要的朋友們可以跟著學(xué)習(xí)下。2021-10-10
本人用 Python 快兩年了，平時只是用它來寫點小小的分析腳本，以方便快捷為主，也沒怎么考慮代碼效率問題。最近想給自己升升級，提高一下代碼的檔次。于是找了一堆效率相關(guān)的文章，做了些實驗，總結(jié)一下。2014-07-07
這篇文章主要介紹了詳解如何修改python中字典的鍵和值，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2020-09-09
詳談套接字中SO_REUSEPORT和SO_REUSEADDR的區(qū)別
下面小編就為大家分享一篇詳談套接字中SO_REUSEPORT和SO_REUSEADDR的區(qū)別，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-04-04
下面小編就為大家分享一篇python 矩陣增加一行或一列的實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-04-04
Python 數(shù)據(jù)處理庫 pandas 入門教程基本操作
pandas是一個Python語言的軟件包，在我們使用Python語言進(jìn)行機器學(xué)習(xí)編程的時候，這是一個非常常用的基礎(chǔ)編程庫。本文是對Python 數(shù)據(jù)處理庫 pandas 入門教程,非常不錯，感興趣的朋友一起看看吧2018-04-04
selenium在執(zhí)行phantomjs的API并獲取執(zhí)行結(jié)果的方法
今天小編就為大家分享一篇selenium在執(zhí)行phantomjs的API并獲取執(zhí)行結(jié)果的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-12-12
python數(shù)據(jù)結(jié)構(gòu)之圖的實現(xiàn)方法
這篇文章主要介紹了python數(shù)據(jù)結(jié)構(gòu)之圖的實現(xiàn)方法,實例分析了Python圖的表示方法與常用尋路算法的實現(xiàn)技巧,需要的朋友可以參考下2015-07-07
python使用pandas自動化合并Excel文件的實現(xiàn)方法
在數(shù)據(jù)分析和處理工作中,經(jīng)常會遇到需要合并多個Excel文件的情況,本文介紹了一種使用Python編程語言中的Pandas庫和Glob模塊來自動化合并Excel文件的方法,需要的朋友可以參考下2024-06-06