pytorch逐元素比较tensor大小实例


Posted in Python onJanuary 03, 2020

如下所示:

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]])

我们取出tenor 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大小实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
在Python的Django框架的视图中使用Session的方法
Jul 23 Python
python中map()与zip()操作方法
Feb 27 Python
Python 迭代器与生成器实例详解
May 18 Python
教你利用Python玩转histogram直方图的五种方法
Jul 30 Python
pygame游戏之旅 添加icon和bgm音效的方法
Nov 21 Python
在pycharm中设置显示行数的方法
Jan 16 Python
Python除法之传统除法、Floor除法及真除法实例详解
May 23 Python
Django框架orM与自定义SQL语句混合事务控制操作
Jun 27 Python
Django框架模型简单介绍与使用分析
Jul 18 Python
python3.6中@property装饰器的使用方法示例
Aug 17 Python
Java Spring项目国际化(i18n)详细方法与实例
Mar 20 Python
用 python 进行微信好友信息分析
Nov 28 Python
pytorch 改变tensor尺寸的实现
Jan 03 #Python
Pytorch Tensor 输出为txt和mat格式方式
Jan 03 #Python
CentOS7下安装python3.6.8的教程详解
Jan 03 #Python
Python实现大数据收集至excel的思路详解
Jan 03 #Python
Python如何基于rsa模块实现非对称加密与解密
Jan 03 #Python
PyTorch的自适应池化Adaptive Pooling实例
Jan 03 #Python
pytorch torch.nn.AdaptiveAvgPool2d()自适应平均池化函数详解
Jan 03 #Python
You might like
Laravel 验证码认证学习记录小结
2019/12/20 PHP
多广告投放代码 推荐
2006/11/13 Javascript
跨浏览器通用、可重用的选项卡tab切换js代码
2011/09/20 Javascript
textarea焦点的用法实现获取焦点清空失去焦点提示效果
2014/05/19 Javascript
用简洁的jQuery方法toggleClass实现隔行换色
2014/10/22 Javascript
Javascript中的五种数据类型详解
2014/12/26 Javascript
js在指定位置增加节点函数insertBefore()用法实例
2015/01/12 Javascript
JS实用技巧小结(屏蔽错误、div滚动条设置、背景图片位置等)
2016/06/16 Javascript
JavaScript 链式结构序列化详解
2016/09/30 Javascript
利用js给datalist或select动态添加option选项的方法
2018/01/25 Javascript
vue用Object.defineProperty手写一个简单的双向绑定的示例
2018/07/09 Javascript
jquery实现联想词搜索框和搜索结果分页的示例
2018/10/10 jQuery
如何通过vscode运行调试javascript代码
2020/07/24 Javascript
JavaScript实现矩形块大小任意缩放
2020/08/25 Javascript
[05:31]DOTA2英雄梦之声_第04期_光之守卫
2014/06/23 DOTA
python网络爬虫采集联想词示例
2014/02/11 Python
python获取文件版本信息、公司名和产品名的方法
2014/10/05 Python
python中logging库的使用总结
2017/10/18 Python
python使用opencv驱动摄像头的方法
2018/08/03 Python
一百行python代码将图片转成字符画
2021/02/19 Python
python的继承知识点总结
2018/12/10 Python
python批量创建指定名称的文件夹
2019/03/21 Python
Python3.5内置模块之shelve模块、xml模块、configparser模块、hashlib、hmac模块用法分析
2019/04/27 Python
python中open函数的基本用法示例
2019/09/07 Python
美国男女折扣服饰百货连锁店:Stein Mart
2017/05/02 全球购物
Gap工厂店:Gap Factory
2017/11/02 全球购物
Sisley法国希思黎美国官方网站:享誉全球的奢华植物美容品牌
2020/06/27 全球购物
金山毒霸系列的笔试题
2013/04/13 面试题
百度JavaScript笔试题
2015/01/15 面试题
小班上学期评语
2014/05/05 职场文书
高中班级口号
2014/06/09 职场文书
英语分层教学实施方案
2014/06/15 职场文书
事业单位个人查摆问题及整改措施
2014/10/28 职场文书
电影建党伟业观后感
2015/06/01 职场文书
Requests什么的通通爬不了的Python超强反爬虫方案!
2021/05/20 Python
如何用vue实现网页截图你知道吗
2021/11/17 Vue.js