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