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操作CouchDB数据库简单示例
Mar 10 Python
python uuid模块使用实例
Apr 08 Python
python实现指定字符串补全空格的方法
Apr 30 Python
Python写的一个简单监控系统
Jun 19 Python
python抓取并保存html页面时乱码问题的解决方法
Jul 01 Python
django数据模型(Model)的字段类型解析
Dec 25 Python
在python中实现求输出1-3+5-7+9-......101的和
Apr 02 Python
Python实现Keras搭建神经网络训练分类模型教程
Jun 12 Python
python 实现两个npy档案合并
Jul 01 Python
Python读写压缩文件的方法
Jul 30 Python
python 中的9个实用技巧,助你提高开发效率
Aug 30 Python
ROS系统将python包编译为可执行文件的简单步骤
Jul 25 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
非常好的php目录导航文件代码
2006/10/09 PHP
php 高效率写法 推荐
2010/02/21 PHP
ThinkPHP3.1新特性之内容解析输出详解
2014/06/19 PHP
PHP实现CSV文件的导入和导出类
2015/03/24 PHP
php 解决substr()截取中文字符乱码问题
2016/07/18 PHP
ExtJS 简介 让你知道extjs是什么
2008/12/29 Javascript
JS 时间显示效果代码
2009/08/23 Javascript
JavaScript Accessor实现说明
2010/12/06 Javascript
open 动态修改img的onclick事件示例代码
2013/11/13 Javascript
jQuery之ajax删除详解
2014/02/27 Javascript
JS判断移动端访问设备并加载对应CSS样式
2014/06/13 Javascript
JS实现向表格行添加新单元格的方法
2015/03/30 Javascript
Jquery遍历Json数据的方法
2015/04/20 Javascript
JavaScript的Backbone.js框架环境搭建及Hellow world示例
2016/05/07 Javascript
Node学习记录之cluster模块
2017/05/31 Javascript
实现单层json按照key字母顺序排序的示例
2017/12/06 Javascript
js实现课堂随机点名系统
2019/11/21 Javascript
jstree中的checkbox默认选中和隐藏示例代码
2019/12/29 Javascript
JS+CSS实现3D切割轮播图
2020/03/21 Javascript
jQuery事件模型默认行为执行顺序及trigger()与 triggerHandler()比较实例分析
2020/04/30 jQuery
调试Python程序代码的几种方法总结
2015/04/28 Python
Python快速从注释生成文档的方法
2016/12/26 Python
python中实现迭代器(iterator)的方法示例
2017/01/19 Python
Python利用turtle库绘制彩虹代码示例
2017/12/20 Python
Python中循环引用(import)失败的解决方法
2018/04/22 Python
Python3中函数参数传递方式实例详解
2019/05/05 Python
pandas针对excel处理的实现
2021/01/15 Python
李宁官方网店:中国运动品牌
2017/11/02 全球购物
M.M.LaFleur官网:美国职业女装品牌
2020/10/27 全球购物
自我评价的范文
2014/02/02 职场文书
农业局学习党的群众路线教育实践活动心得体会
2014/03/07 职场文书
小学生我的梦想演讲稿
2014/08/21 职场文书
毕业设计论文评语
2014/12/31 职场文书
海洋天堂观后感
2015/06/05 职场文书
适合青年人白手起家的创业项目分享
2019/08/16 职场文书
Ubuntu Server 安装Tomcat并配置systemctl
2022/04/28 Servers