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图形用户界面tkinter之按钮Button的使用说明
这篇文章主要介绍了python图形用户界面tkinter之按钮Button的使用说明，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2022-06-06
这篇文章主要介绍了Django ORM 查询表中某列字段值的方法，本文通过实例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下2020-04-04
PyCharm中New Directory 和 New Python
python package这是一个特殊的目录,因为在创建该python package的时候,系统会自动地生成一个py文件, init.py,这篇文章主要介绍了PyCharm中New Directory 和 New Python Package的区别,需要的朋友可以参考下2023-12-12
今天小编就为大家分享一篇Python一句代码实现找出所有水仙花数的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-11-11
这篇文章主要给大家介绍了关于Django跨域请求问题解决的相关资料，文中介绍的实现方法包括：使用django-cors-headers全局控制、使用JsonP，只能用于Get方法以及在views.py里设置响应头，只能控制单个接口，需要的朋友可以参考下2018-06-06
这篇文章主要为大家详细介绍了如何利用Python和Pygame实现24点小游戏，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2022-04-04
下面小编就为大家分享一篇python去除扩展名的实例讲解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-04-04
这篇文章主要介绍了NumPy-ndarray 的数据类型用法说明，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2021-05-05
本文介绍了Python中列表的逆序、复制和清除操作,通过reverse()方法、切片、copy()方法和clear()方法,我们可以轻松地对列表进行这些操作2024-12-12
这篇文章主要介绍了python返回昨天日期的方法,涉及Python日期操作的相关技巧,需要的朋友可以参考下2015-05-05
最新评论