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]相當于[:, :, … :, 0]
以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
最近接觸了個項目，需要把多個文件合成一個視頻，本文主要使用Python把多個MP4合成視頻，感興趣的可以了解一下2021-07-07
這篇文章主要介紹了python如何繪制極坐標輪廓圖contourf問題,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2023-08-08
本文主要介紹了 Python賦值邏輯的實現(xiàn)，文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2023-02-02
python+webdriver自動化環(huán)境搭建步驟詳解
在本篇文章里小編給大家分享了關(guān)于python+webdriver自動化環(huán)境搭建的詳細步驟以及注意點，需要的朋友們參考下。2019-06-06
這篇文章主要介紹了Python SSL證書驗證問題解決方案,文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2020-01-01
很多人在創(chuàng)作視頻時都需要進行剪輯,本文主要介紹了Python視頻剪輯合并操作的實現(xiàn)示例,文中通過示例代碼介紹的非常詳細,對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2025-01-01
Python圖形化界面基礎(chǔ)篇之如何使用彈出窗口和對話框
對于Python程序員來說,處理彈出窗口似乎并不是一個常見的任務(wù),這篇文章主要給大家介紹了關(guān)于Python圖形化界面基礎(chǔ)篇之如何使用彈出窗口和對話框的相關(guān)資料,需要的朋友可以參考下2024-03-03
這篇文章主要給大家介紹了關(guān)于Python第三方庫OS庫的相關(guān)資料,os庫主要是對文件和文件夾進行操作,在Python中對?件和?件夾的操作要借助os模塊??的相關(guān)功能,需要的朋友可以參考下2024-06-06
pycharm中出現(xiàn)no module named xlwt的原因及解決
這篇文章主要介紹了pycharm中出現(xiàn)no module named xlwt的原因及解決方案，具有很好的參考價值，希望對大家有所幫助。2023-05-05
使用Python實現(xiàn)XLS和XLSX之間的相互轉(zhuǎn)換
在日常工作中,我們經(jīng)常需要處理和轉(zhuǎn)換不同格式的Excel文件,以適應(yīng)不同的需求和軟件兼容性,Excel文件的兩種常見格式是XLS（Excel 97-2003）和XLSX（Excel 2007及以上版本）,本文將詳細介紹如何使用Python在XLS和XLSX格式之間進行轉(zhuǎn)換,需要的朋友可以參考下2024-09-09