最新下载
热门教程
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
pytorch逐元素比较tensor大小代码实例
时间:2020-01-04 编辑:袖梨 来源:一聚教程网
本篇文章小编给大家分享一下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]])
相关文章
- Golang ProtoBuf的基本语法详解 10-20
- Python识别MySQL中的冗余索引解析 10-20
- Python+Pygame绘制小球代码展示 10-18
- Python中的数据精度问题介绍 10-18
- Python随机值生成的常用方法介绍 10-18
- python3解压缩.gz文件分析 09-27