pytorch逐元素比較tensor大小實例
更新時間：2020年01月03日 15:00:00 作者：MD2017
今天小編就為大家分享一篇pytorch逐元素比較tensor大小實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
如下所示：
import torch a = torch.tensor([[0.01, 0.011], [0.009, 0.9]]) mask = a.gt(0.01) print(mask)
tensor比較大小可以用tensor.gt屬性。上面比較了a中每個元素和0.01的大小，大于0.01的元素輸出True。輸出結果：
tensor([[False, True],
[False, True]])
我們?nèi)〕鰐enor a中對應的大于0.01的值：
a[mask]
將對應滿足條件的元素輸出并自動拉伸為一個一維向量輸出：
tensor([0.0110, 0.9000])
我們也可以比較兩個tensor大小
b = torch.tensor([[0.02, 1], [0, 1.0]])
torch.gt(a, b)
tensor([[False, False],
[ True, False]])
以上這篇pytorch逐元素比較tensor大小實例就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
這篇文章主要介紹了python 使用pandas讀取csv文件的方法,本文結合示例代碼給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值，需要的朋友可以參考下2022-12-12
Python Print實現(xiàn)在輸出中插入變量的例子
今天小編就為大家分享一篇Python Print實現(xiàn)在輸出中插入變量的例子，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-12-12
了解一下python內(nèi)建模塊collections
這篇文章主要介紹了Python內(nèi)建模塊——collections的相關資料，幫助大家更好的理解和使用python，感興趣的朋友可以了解下2020-09-09
這篇文章主要為大家詳細介紹了Python之Selenium自動化爬蟲，文中示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下，希望能夠給你帶來幫助2022-01-01
Jupyter?Notebook是一個開源的交互式編程環(huán)境，用于創(chuàng)建和共享包含實時代碼、文本等，本文主要來和大家分享一些有趣的Jupyter?Notebook魔法命令，需要的可以參考一下2023-07-07
Python使用paramiko連接遠程服務器執(zhí)行Shell命令的實現(xiàn)
這篇文章主要介紹了Python使用paramiko連接遠程服務器執(zhí)行Shell命令的實現(xiàn),本文給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值，需要的朋友可以參考下2021-03-03
python實現(xiàn)圖片加文字水印OPenCV和PIL庫
本文來為大家介紹一下，使用python中的庫實現(xiàn)給圖片添加文字水印，openCV可以給圖片添加水印，如果要添加漢字水印那就要使用PIL庫2021-09-09
在本篇文章當中主要給大家深入介紹在?cpython?當中非常重要的一個數(shù)據(jù)結構?code?object!?我們簡單介紹了一下在?code?object?當中有哪些字段以及這些字段的簡單含義，在本篇文章當中將會舉一些例子以便更加深入理解這些字段2023-04-04
這篇文章主要介紹了Python?中enum的使用方法總結，枚舉在許多編程語言中常被表示為一種基礎的數(shù)據(jù)結構使用，下文更多詳細內(nèi)容需要的小伙伴可以參考一下2022-03-03
這篇文章主要介紹了Java中重定向輸出流實現(xiàn)用文件記錄程序日志,本文直接給出代碼實例,并對代碼做了詳細注解,需要的朋友可以參考下2015-06-06