pytorch 實現(xiàn)刪除tensor中的指定行列
更新時間：2020年01月13日 09:18:42 作者：qq_29007291
今天小編就為大家分享一篇pytorch 實現(xiàn)刪除tensor中的指定行列，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
前言
在pytorch中， 想刪除tensor中的指定行列，原本以為有個函數(shù)或者直接把某一行賦值為[]就可以，結(jié)果發(fā)現(xiàn)沒這么簡單，因此用了一個曲線救國方法，希望如果有更直接的方法，請大家指出。
code
本質(zhì)上是利用mask刪除了指定行，然后重新指向。
a = torch.rand(4, 2)
print(a)
idx = 1
a = a[torch.arange(a.size(0))!=1]
print(a)
"""
tensor([[2.7775e-01, 3.7430e-01],
[9.0373e-01, 8.1220e-02],
[9.8638e-01, 8.6293e-01],
[9.8139e-04, 9.8460e-02]])
tensor([[2.7775e-01, 3.7430e-01],
[9.8638e-01, 8.6293e-01],
[9.8139e-04, 9.8460e-02]])
"""
以上這篇pytorch 實現(xiàn)刪除tensor中的指定行列就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
下面小編就為大家?guī)硪黄唵握務(wù)刾ython中的語句和語法。小編覺得挺不錯的，現(xiàn)在就分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2017-08-08
這篇文章主要介紹了用Python實現(xiàn)一個簡單的線程池,通過這個小程序可以幫助更好地理解Python中線程的運行機制,需要的朋友可以參考下2015-04-04
這篇文章主要為大家詳細介紹了Python實現(xiàn)LRU緩存置換算法，文中示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下2022-05-05
本篇文章給大家分享了用python腳本寫出作為Windows服務(wù)啟動功能，對此有需求的朋友跟著小編一起學(xué)習(xí)下。2018-02-02
這篇文章主要介紹了【python】str與json類型轉(zhuǎn)換，文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2019-04-04
這篇文章主要介紹了使用Python打造一款間諜程序,本文通過實例代碼給大家介紹的非常詳細，具有一定的參考借鑒價值，需要的朋友可以參考下2020-02-02
這篇文章主要給大家介紹了關(guān)于Python求最小公倍數(shù)4種方法的相關(guān)資料,最小公倍數(shù)不可以像最大公約數(shù)那樣直接利用輾轉(zhuǎn)相除法求出,但可以借助輾轉(zhuǎn)相除法求得的最大公約數(shù)來求最小公倍數(shù),需要的朋友可以參考下2023-10-10
這篇文章主要介紹了Python中協(xié)程用法代碼詳解，分享了相關(guān)代碼示例，小編覺得還是挺不錯的，具有一定借鑒價值，需要的朋友可以參考下2018-02-02
解決ImportError:cannot import name ‘Flatten‘&nb
這篇文章主要介紹了解決ImportError:cannot import name ‘Flatten‘ from ‘torch.nn‘問題，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2023-06-06
這篇文章主要介紹了Python自動化測試selenium指定截圖文件名方法,Selenium 支持 Web 瀏覽器的自動化,它提供一套測試函數(shù),用于支持 Web 自動化測試,下文基于python實現(xiàn)指定截圖文件名方法,需要的小伙伴可以參考一下2025-02-02