pytorch逐元素比较tensor大小实例
更新时间：2020年01月03日 15:00:00 作者：MD2017
今天小编就为大家分享一篇pytorch逐元素比较tensor大小实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
如下所示：
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 使用pandas读取csv文件的方法,本文结合示例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2022-12-12
今天小编就为大家分享一篇Python Print实现在输出中插入变量的例子，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-12-12
这篇文章主要介绍了Python内建模块——collections的相关资料，帮助大家更好的理解和使用python，感兴趣的朋友可以了解下2020-09-09
这篇文章主要为大家详细介绍了Python之Selenium自动化爬虫，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下，希望能够给你带来帮助2022-01-01
Jupyter Notebook是一个开源的交互式编程环境，用于创建和共享包含实时代码、文本等，本文主要来和大家分享一些有趣的Jupyter Notebook魔法命令，需要的可以参考一下2023-07-07
Python使用paramiko连接远程服务器执行Shell命令的实现
这篇文章主要介绍了Python使用paramiko连接远程服务器执行Shell命令的实现,本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2021-03-03
本文来为大家介绍一下，使用python中的库实现给图片添加文字水印，openCV可以给图片添加水印，如果要添加汉字水印那就要使用PIL库2021-09-09
在本篇文章当中主要给大家深入介绍在 cpython 当中非常重要的一个数据结构 code object! 我们简单介绍了一下在 code object 当中有哪些字段以及这些字段的简单含义，在本篇文章当中将会举一些例子以便更加深入理解这些字段2023-04-04
这篇文章主要介绍了Python 中enum的使用方法总结，枚举在许多编程语言中常被表示为一种基础的数据结构使用，下文更多详细内容需要的小伙伴可以参考一下2022-03-03
这篇文章主要介绍了Java中重定向输出流实现用文件记录程序日志,本文直接给出代码实例,并对代码做了详细注解,需要的朋友可以参考下2015-06-06
最新评论