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根軸上的第一列數字組成tensor,[0, …]表示抽取tensor b的第一根軸上的第一列數字組成tensor,[0, …, 0]表示抽取b的第一根和第四根軸上的第一列數字組成tensor。
還發現一個現象
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]
以上為個人經驗,希望能給大傢一個參考,也希望大傢多多支持WalkonNet。
推薦閱讀:
- Broadcast廣播機制在Pytorch Tensor Numpy中的使用詳解
- pytorch中的廣播語義
- Pytorch中torch.stack()函數的深入解析
- pytorch教程之Tensor的值及操作使用學習
- Python深度學習之Pytorch初步使用