pytorch 實現(xiàn)刪除tensor中的指定行列
更新時間：2020年01月13日 09:18:42 作者：qq_29007291
今天小編就為大家分享一篇pytorch 實現(xiàn)刪除tensor中的指定行列，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
前言
在pytorch中， 想刪除tensor中的指定行列，原本以為有個函數(shù)或者直接把某一行賦值為[]就可以，結果發(fā)現(xiàn)沒這么簡單，因此用了一個曲線救國方法，希望如果有更直接的方法，請大家指出。
code
本質上是利用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中的指定行列就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
這篇文章主要介紹了Python實現(xiàn)的簡單萬年歷例子分享,需要的朋友可以參考下2014-04-04
Python 實現(xiàn)的 Google 批量翻譯功能
這篇文章主要介紹了Python 實現(xiàn)的 Google 批量翻譯功能，非常不錯，具有一定的參考借鑒價值,需要的朋友可以參考下2019-08-08
這篇文章主要介紹了使用python請求接口方式(可進行并發(fā)測試),具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2024-06-06
大麥網(wǎng)，是中國綜合類現(xiàn)場娛樂票務營銷平臺，業(yè)務覆蓋演唱會、 話劇、音樂劇、體育賽事等領域。但是因為票數(shù)有限，還有黃牛們不能丟了飯碗，所以導致了，很多人都搶不到票，那么，今天帶大家用Python來制作一個自動搶票的腳本小程序2021-10-10
python中的一些類型轉換函數(shù)，供大家參考2013-02-02
python opencv實現(xiàn)圖片缺陷檢測（講解直方圖以及相關系數(shù)對比法）
這篇文章主要介紹了python opencv 圖片缺陷檢測，主要講解直方圖以及相關系數(shù)對比法，本文通過實例截圖給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值,需要的朋友可以參考下2020-04-04
這篇文章主要介紹了Python 多線程之threading 模塊的使用，幫助大家更好的理解和學習使用python，感興趣的朋友可以了解下2021-04-04
python3.8與pyinstaller沖突問題的快速解決方法
這篇文章主要介紹了python3.8與pyinstaller沖突問題及解決方法，本文給大家介紹的非常詳細，具有一定的參考借鑒價值,需要的朋友可以參考下2020-01-01
使用Python編寫類UNIX系統(tǒng)的命令行工具的教程
這篇文章主要介紹了使用Python編寫類UNIX系統(tǒng)的命令行工具的教程,本文來自于IBM官方網(wǎng)站技術文檔,需要的朋友可以參考下2015-04-04
Pandas是基于NumPy的一種工具,該工具是為了解決數(shù)據(jù)分析任務而創(chuàng)建的,本文主要介紹了pandas DataFrame.to_sql()用法小結,感興趣的可以了解一下2024-02-02