以二維numpy矩陣為例
import torch
import numpy as np
K=3 #取每行最小3個(gè)值的索引
data=np.random.rand(4,7)
print(data)
data=torch.from_numpy(data)
a, idx = torch.sort(data, descending=False)
lists=idx[:,:K]
print(lists)
運(yùn)行結(jié)果如下：
results.jpg
pytorch 計(jì)算2維矩陣每行最小N個(gè)值的索引
最后編輯于 ：
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
- 文/潘曉璐 我一進(jìn)店門冻记，熙熙樓的掌柜王于貴愁眉苦臉地迎上來(lái)，“玉大人来惧，你說(shuō)我怎么就攤上這事冗栗。” “怎么了供搀？”我有些...
- 文/不壞的土叔 我叫張陵隅居，是天一觀的道長(zhǎng)。 經(jīng)常有香客問(wèn)我葛虐，道長(zhǎng)胎源，這世上最難降的妖魔是什么？ 我笑而不...
- 正文 為了忘掉前任屿脐，我火速辦了婚禮涕蚤，結(jié)果婚禮上，老公的妹妹穿的比我還像新娘的诵。我一直安慰自己万栅，他們只是感情好，可當(dāng)我...
- 文/花漫 我一把揭開白布奢驯。 她就那樣靜靜地躺著申钩，像睡著了一般。 火紅的嫁衣襯著肌膚如雪瘪阁。 梳的紋絲不亂的頭發(fā)上撒遣，一...
- 那天，我揣著相機(jī)與錄音管跺，去河邊找鬼义黎。 笑死，一個(gè)胖子當(dāng)著我的面吹牛豁跑，可吹牛的內(nèi)容都是我干的廉涕。 我是一名探鬼主播，決...
- 文/蒼蘭香墨 我猛地睜開眼，長(zhǎng)吁一口氣：“原來(lái)是場(chǎng)噩夢(mèng)啊……” “哼狐蜕！你這毒婦竟也來(lái)了宠纯？” 一聲冷哼從身側(cè)響起，我...
- 序言：老撾萬(wàn)榮一對(duì)情侶失蹤层释，失蹤者是張志新（化名）和其女友劉穎婆瓜，沒想到半個(gè)月后，有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體贡羔，經(jīng)...
- 正文 獨(dú)居荒郊野嶺守林人離奇死亡廉白，尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
- 正文 我和宋清朗相戀三年，在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了乖寒。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片猴蹂。...
- 正文 年R本政府宣布瓢省，位于F島的核電站，受9級(jí)特大地震影響痊班，放射性物質(zhì)發(fā)生泄漏勤婚。R本人自食惡果不足惜，卻給世界環(huán)境...
- 文/蒙蒙 一涤伐、第九天 我趴在偏房一處隱蔽的房頂上張望馒胆。 院中可真熱鬧，春花似錦凝果、人聲如沸祝迂。這莊子的主人今日做“春日...
- 文/蒼蘭香墨 我抬頭看了看天上的太陽(yáng)型雳。三九已至，卻和暖如春山害，著一層夾襖步出監(jiān)牢的瞬間纠俭，已是汗流浹背。 一陣腳步聲響...
- 正文 我出身青樓权纤，卻偏偏與公主長(zhǎng)得像钓简，于是被迫代替她去往敵國(guó)和親乌妒。 傳聞我的和親對(duì)象是個(gè)殘疾皇子，可洞房花燭夜當(dāng)晚...
推薦閱讀更多精彩內(nèi)容
- 本章主要介紹兩個(gè)內(nèi)容外邓，2.1節(jié)介紹如何安裝PyTorch撤蚊，以及如何配置學(xué)習(xí)環(huán)境；2.2節(jié)將帶領(lǐng)讀者快速瀏覽PyTo...
- 我們先來(lái)看一首詩(shī)损话。 深宮有奇物拴魄，璞玉冠何有。度歲忽如何席镀，遐齡復(fù)何欲。學(xué)來(lái)玉階上夏漱，仰望金閨籍豪诲。習(xí)協(xié)萬(wàn)壑間，高高萬(wàn)象逼...
- 看《深度學(xué)習(xí)圖解》的第6章的代碼挂绰，前后對(duì)于矩陣的索引操作代碼略有區(qū)別屎篱。心中疑問(wèn)重重，為何兩者都能工作葵蒂。帶著疑問(wèn)交播，網(wǎng)...
- 2.1 環(huán)境配置[https://tangshusen.me/Dive-into-DL-PyTorch/#/cha...
- Numpy的安裝 驗(yàn)證安裝。 使用隨機(jī)命令random.rand(4,4)生成一個(gè)4乘4矩陣践付。 Numpy的基本使...
