主頁 > 知識庫 > pytorch中[..., 0]的用法說明

pytorch中[..., 0]的用法說明

熱門標簽:池州外呼調研線路 江西省地圖標注 武漢外呼系統(tǒng)平臺 富錦商家地圖標注 沈陽防封電銷卡品牌 沈陽人工外呼系統(tǒng)價格 如何申請400電話費用 沈陽外呼系統(tǒng)呼叫系統(tǒng) 外呼系統(tǒng)哪些好辦

在看程序的時候看到了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)

結果為True

2. 示例2

import torch
a = torch.rand((64, 17, 24, 8))
b = a[..., 0]
c = a[:, :, :, 0]
print(b == c)

結果為True

3. 結論

可以看出[…, 0]相當于[:, :, … :, 0]

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

您可能感興趣的文章:
  • pytorch中的nn.ZeroPad2d()零填充函數(shù)實例詳解
  • 基于python及pytorch中乘法的使用詳解
  • PyTorch安裝與基本使用詳解

標簽:黑龍江 常德 銅川 呂梁 潛江 阿里 株洲 通遼

巨人網(wǎng)絡通訊聲明:本文標題《pytorch中[..., 0]的用法說明》,本文關鍵詞  pytorch,中,...,的,用法,說明,;如發(fā)現(xiàn)本文內容存在版權問題,煩請?zhí)峁┫嚓P信息告之我們,我們將及時溝通與處理。本站內容系統(tǒng)采集于網(wǎng)絡,涉及言論、版權與本站無關。
  • 相關文章
  • 下面列出與本文章《pytorch中[..., 0]的用法說明》相關的同類信息!
  • 本頁收集關于pytorch中[..., 0]的用法說明的相關信息資訊供網(wǎng)民參考!
  • 推薦文章