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如何重载模块实例解析
Jan 25 Python
Python异常处理操作实例详解
Aug 28 Python
使用Python制作自动推送微信消息提醒的备忘录功能
Sep 06 Python
python 2.7.13 安装配置方法图文教程
Sep 18 Python
python执行CMD指令,并获取返回的方法
Dec 19 Python
Python多版本开发环境管理工具介绍
Jul 03 Python
python用for循环求和的方法总结
Jul 08 Python
python并发编程多进程 互斥锁原理解析
Aug 20 Python
python模块导入的方法
Oct 24 Python
keras小技巧——获取某一个网络层的输出方式
May 23 Python
keras模型保存为tensorflow的二进制模型方式
May 25 Python
Pytorch之扩充tensor的操作
Mar 04 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自动跳转中英文页面
2008/07/29 PHP
PHP+Javascript实现在线拍照功能实例
2015/07/18 PHP
php给图片加文字水印
2015/07/31 PHP
WordPress中用于创建以及获取侧边栏的PHP函数讲解
2015/12/29 PHP
thinkPHP框架中执行事务的方法示例
2018/05/31 PHP
jquery实现背景墙聚光灯效果示例分享
2014/03/02 Javascript
JavaScript极简入门教程(三):数组
2014/10/25 Javascript
jquery实现相册一下滑动两次的方法
2015/02/09 Javascript
60行js代码实现俄罗斯方块
2015/03/31 Javascript
jQuery中ajax的load()与post()方法实例详解
2016/01/05 Javascript
基于原生js淡入淡出函数封装(兼容IE)
2016/10/20 Javascript
详解js的事件代理(委托)
2016/12/22 Javascript
详解Angualr 组件间通信
2017/01/21 Javascript
微信小程序 向左滑动删除功能的实现
2017/03/10 Javascript
vue-cli3+ts+webpack实现多入口多出口功能
2019/05/30 Javascript
Vue中Table组件行内右键菜单实现方法(基于 vue + AntDesign)
2019/11/21 Javascript
javascript实现移动端红包雨页面
2020/06/23 Javascript
解决父组件将子组件作为弹窗调用只执行一次created的问题
2020/07/24 Javascript
Javascript如何递归遍历本地文件夹
2020/08/06 Javascript
Python守护进程用法实例分析
2015/06/04 Python
Python标准库之itertools库的使用方法
2017/09/07 Python
利用Python如何将数据写到CSV文件中
2018/06/05 Python
Python的iOS自动化打包实例代码
2018/11/22 Python
DRF跨域后端解决之django-cors-headers的使用
2019/01/27 Python
Python使用OpenPyXL处理Excel表格
2020/07/02 Python
pycharm激活码2020最新分享适用pycharm2020最新版亲测可用
2020/11/22 Python
Python实现钉钉/企业微信自动打卡的示例代码
2021/02/02 Python
canvas 实现 github404动态效果的示例代码
2017/11/15 HTML / CSS
GC是什么?为什么要有GC?
2013/12/08 面试题
视光学毕业生自荐书范文
2014/02/13 职场文书
学雷锋树新风演讲稿
2014/05/10 职场文书
开展批评与自我批评发言材料
2014/10/17 职场文书
2014年保管员工作总结
2014/11/18 职场文书
通知的写法
2015/04/23 职场文书
2015中秋节晚会主持词
2015/07/01 职场文书
MYSQL事务的隔离级别与MVCC
2022/05/25 MySQL