pytorch逐元素比較tensor大小實(shí)例
更新時(shí)間：2020年01月03日 15:00:00 作者：MD2017
今天小編就為大家分享一篇pytorch逐元素比較tensor大小實(shí)例，具有很好的參考價(jià)值，希望對大家有所幫助。一起跟隨小編過來看看吧
如下所示：
import torch a = torch.tensor([[0.01, 0.011], [0.009, 0.9]]) mask = a.gt(0.01) print(mask)
tensor比較大小可以用tensor.gt屬性。上面比較了a中每個(gè)元素和0.01的大小，大于0.01的元素輸出True。輸出結(jié)果：
tensor([[False, True],
[False, True]])
我們?nèi)〕鰐enor a中對應(yīng)的大于0.01的值：
a[mask]
將對應(yīng)滿足條件的元素輸出并自動拉伸為一個(gè)一維向量輸出：
tensor([0.0110, 0.9000])
我們也可以比較兩個(gè)tensor大小
b = torch.tensor([[0.02, 1], [0, 1.0]])
torch.gt(a, b)
tensor([[False, False],
[ True, False]])
以上這篇pytorch逐元素比較tensor大小實(shí)例就是小編分享給大家的全部內(nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
python圖形用戶界面tkinter之按鈕Button的使用說明
這篇文章主要介紹了python圖形用戶界面tkinter之按鈕Button的使用說明，具有很好的參考價(jià)值，希望對大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教2022-06-06
這篇文章主要介紹了Django ORM 查詢表中某列字段值的方法，本文通過實(shí)例代碼給大家介紹的非常詳細(xì)，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2020-04-04
PyCharm中New Directory 和 New Python
python package這是一個(gè)特殊的目錄,因?yàn)樵趧?chuàng)建該python package的時(shí)候,系統(tǒng)會自動地生成一個(gè)py文件, init.py,這篇文章主要介紹了PyCharm中New Directory 和 New Python Package的區(qū)別,需要的朋友可以參考下2023-12-12
Python一句代碼實(shí)現(xiàn)找出所有水仙花數(shù)的方法
今天小編就為大家分享一篇Python一句代碼實(shí)現(xiàn)找出所有水仙花數(shù)的方法，具有很好的參考價(jià)值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-11-11
這篇文章主要給大家介紹了關(guān)于Django跨域請求問題解決的相關(guān)資料，文中介紹的實(shí)現(xiàn)方法包括：使用django-cors-headers全局控制、使用JsonP，只能用于Get方法以及在views.py里設(shè)置響應(yīng)頭，只能控制單個(gè)接口，需要的朋友可以參考下2018-06-06
Python+Pygame實(shí)戰(zhàn)之24點(diǎn)游戲的實(shí)現(xiàn)
這篇文章主要為大家詳細(xì)介紹了如何利用Python和Pygame實(shí)現(xiàn)24點(diǎn)小游戲，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2022-04-04
下面小編就為大家分享一篇python去除擴(kuò)展名的實(shí)例講解，具有很好的參考價(jià)值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-04-04
NumPy-ndarray 的數(shù)據(jù)類型用法說明
這篇文章主要介紹了NumPy-ndarray 的數(shù)據(jù)類型用法說明，具有很好的參考價(jià)值，希望對大家有所幫助。一起跟隨小編過來看看吧2021-05-05
Python中實(shí)現(xiàn)列表的逆序、復(fù)制與清除的幾種常見方法
本文介紹了Python中列表的逆序、復(fù)制和清除操作,通過reverse()方法、切片、copy()方法和clear()方法,我們可以輕松地對列表進(jìn)行這些操作2024-12-12
這篇文章主要介紹了python返回昨天日期的方法,涉及Python日期操作的相關(guān)技巧,需要的朋友可以參考下2015-05-05