pytorch索引查找 index_select的例子
更新時間：2019年08月18日 19:18:23 作者：ShellCollector
今天小編就為大家分享一篇pytorch索引查找 index_select的例子，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
index_select
anchor_w = self.FloatTensor(self.scaled_anchors).index_select(1, self.LongTensor([0]))
參數(shù)說明：index_select(x, 1, indices)
1代表維度1，即列，indices是篩選的索引序號。
例子：
import torch x = torch.linspace(1, 12, steps=12).view(3,4) print(x) indices = torch.LongTensor([0, 2]) y = torch.index_select(x, 0, indices) print(y) z = torch.index_select(x, 1, indices) print(z) z = torch.index_select(y, 1, indices) print(z)
結(jié)果：
tensor([[ 1., 2., 3., 4.],
[ 5., 6., 7., 8.],
[ 9., 10., 11., 12.]])
tensor([[ 1., 2., 3., 4.],
[ 9., 10., 11., 12.]])
tensor([[ 1., 3.],
[ 5., 7.],
[ 9., 11.]])
tensor([[ 1., 3.],
[ 9., 11.]])
以上這篇pytorch索引查找 index_select的例子就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
本文主要介紹了Appium自動化測試實現(xiàn)H5頁面元素定位，文中通過示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下2022-02-02
使用PyTorch將數(shù)據(jù)從CPU移動到GPU的四個方法
這篇文章給大家介紹了在 PyTorch 中,將數(shù)據(jù)從 CPU 移動到 GPU 的幾種方法,使用 .to() 方法,使用 .cuda() 方法,使用 torch.Tensor 構(gòu)造函數(shù)和使用 torch.tensor 構(gòu)造函數(shù)這四個方法,通過代碼示例介紹非常詳細,需要的朋友可以參考下2024-01-01
python+logging+yaml實現(xiàn)日志分割
這篇文章主要為大家詳細介紹了python+logging+yaml實現(xiàn)日志分割，具有一定的參考價值，感興趣的小伙伴們可以參考一下2019-07-07
這篇文章主要介紹了Python xlwt模塊使用代碼實例,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2020-06-06
Python socket套接字實現(xiàn)C/S模式遠程命令執(zhí)行功能案例
這篇文章主要介紹了Python socket套接字實現(xiàn)C/S模式遠程命令執(zhí)行功能,涉及Python socket套接字編寫服務器/客戶機模式數(shù)據(jù)傳輸相關(guān)操作技巧,需要的朋友可以參考下2018-07-07
python使用xlsxwriter實現(xiàn)有向無環(huán)圖到Excel的轉(zhuǎn)換
這篇文章主要為大家詳細介紹了python使用xlsxwriter實現(xiàn)有向無環(huán)圖到Excel的轉(zhuǎn)換，具有一定的參考價值，感興趣的小伙伴們可以參考一下2018-12-12
tensorflow使用range_input_producer多線程讀取數(shù)據(jù)實例
今天小編就為大家分享一篇tensorflow使用range_input_producer多線程讀取數(shù)據(jù)實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-01-01
這篇文章介紹了Python實現(xiàn)猜數(shù)字小游戲，文中通過示例代碼介紹的非常詳細。對大家的學習或工作具有一定的參考借鑒價值，需要的朋友可以收藏下，方便下次瀏覽觀看2021-12-12
今天小編就為大家分享一篇python實現(xiàn)智能語音天氣預報，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-12-12
Pytorch從0實現(xiàn)Transformer的實踐
本文主要介紹了Pytorch從0實現(xiàn)Transformer的實踐，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2022-05-05