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操作xml文件详细介绍
Jun 09 Python
python使用自定义user-agent抓取网页的方法
Apr 15 Python
编写Python脚本来实现最简单的FTP下载的教程
May 04 Python
Python实现求笛卡尔乘积的方法
Sep 16 Python
python+pandas生成指定日期和重采样的方法
Apr 11 Python
对python插入数据库和生成插入sql的示例讲解
Nov 14 Python
Python 3.3实现计算两个日期间隔秒数/天数的方法示例
Jan 07 Python
Python中print和return的作用及区别解析
May 05 Python
django认证系统 Authentication使用详解
Jul 22 Python
解决Keras 中加入lambda层无法正常载入模型问题
Jun 16 Python
Python用K-means聚类算法进行客户分群的实现
Aug 23 Python
python编写函数注意事项总结
Mar 29 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 CLI实现简单的数据库实时监控调度
2009/07/01 PHP
PHP将整个网站生成HTML纯静态网页的方法总结
2012/02/05 PHP
Zend Framework教程之分发器Zend_Controller_Dispatcher用法详解
2016/03/07 PHP
php实现文件上传及头像预览功能
2017/01/15 PHP
Laravel用户授权系统的使用方法示例
2018/09/16 PHP
PHP中抽象类,接口功能、定义方法示例
2019/02/26 PHP
JMenuTab简单使用说明
2008/03/13 Javascript
3Z版基于jquery的图片复选框(asp.net+jquery)
2010/04/12 Javascript
UserData用法总结 lanyu出品
2010/07/01 Javascript
JQuery记住用户名密码实现下次自动登录功能
2015/04/27 Javascript
jquery实现超简洁的TAB选项卡效果代码
2015/08/28 Javascript
深入浅析JSON.parse()、JSON.stringify()和eval()的作用详解
2016/04/03 Javascript
微信小程序 视图层(xx.xml)和逻辑层(xx.js)详细介绍
2016/10/13 Javascript
bootstrap多种样式进度条展示
2016/12/20 Javascript
两种简单的跨域方法(jsonp、php)
2017/01/02 Javascript
微信小程序 下拉列表的实现实例代码
2017/03/08 Javascript
微信小程序左滑动显示菜单功能的实现
2018/06/14 Javascript
详解JSON Web Token 入门教程
2018/07/30 Javascript
element UI upload组件上传附件格式限制方法
2018/09/04 Javascript
详解IOS微信上Vue单页面应用JSSDK签名失败解决方案
2018/11/14 Javascript
详解如何在JS代码中消灭for循环
2019/12/11 Javascript
vue.js实现双击放大预览功能
2020/06/23 Javascript
python机器学习实战之最近邻kNN分类器
2017/12/20 Python
Centos7 Python3下安装scrapy的详细步骤
2018/03/15 Python
一篇文章读懂Python赋值与拷贝
2018/04/19 Python
selenium + python 获取table数据的示例讲解
2018/10/13 Python
Python脚本修改阿里云的访问控制列表的方法
2019/03/08 Python
Python提取特定时间段内数据的方法实例
2019/04/01 Python
Selenium 配置启动项参数的方法
2020/12/04 Python
CSS3常用的几种颜色渐变模式总结
2016/11/18 HTML / CSS
日本高端护肤品牌:Tatcha
2016/08/29 全球购物
Nginx tp3.2.3 404问题解决方案
2021/03/31 Servers
Java实现二分搜索树的示例代码
2022/03/17 Java/Android
vue 实现弹窗关闭后刷新效果
2022/04/08 Vue.js
Win10 Anaconda安装python-pcl
2022/04/29 Servers
JS前端使用canvas实现扩展物体类和事件派发
2022/08/05 Javascript