使用pytorch 筛选出一定范围的值
更新时间：2020年06月28日 09:59:21 作者：guotong1988
这篇文章主要介绍了使用pytorch 筛选出一定范围的值，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
我就废话不多说了，大家还是直接看代码吧~
import torch input_tensor = torch.tensor([1,2,3,4,5]) print(input_tensor>3) mask = (input_tensor>3).nonzero() print(mask) print(input_tensor.index_select(0,mask))
tensor([0, 0, 0, 1, 1], dtype=torch.uint8) tensor([3, 4]) tensor([4, 5])
补充知识：pytorch tensor筛选满足条件的行或列（使用与或）
我就废话不多说了，大家还是直接看代码吧~
import torch x = torch.linspace(1, 8, steps=8).view(4, 2) print(x) area1=(x[:,0]>5.5)&(x[:,1]>5.5) c=x[:,0]*x[:,1] area2=c>25 area=area1|area2 print(x[area]) if 0: # index=torch.max(area,1)[0] b=x[area] # b= x[torch.where((x[:,0]>0) & (x[:,0]<6))] # print(b)
以上这篇使用pytorch 筛选出一定范围的值就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要介绍了用Python批量处理图片大小尺寸的方法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习吧2022-11-11
Python Fuzzywuzzy库基本函数及模糊字符串匹配应用实战
fuzzywuzzy 是一个用于模糊字符串匹配的 Python 库,它基于编辑距离算法,提供了多个函数来比较字符串之间的相似性,在实际开发中,字符串匹配是一项常见但具有挑战性的任务,用户可能犯拼写错误,使用缩写或者输入同义词,因此,我们需要一种方法来处理这些情况2023-12-12
这篇文章主要为大家介绍了python Exception常规异常基类详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2023-06-06
这篇文章主要介绍了pytorch模型训练的时候GPU使用率不高的问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教2023-09-09
在本篇文章里小编给大家分享的是关于Django分组聚合查询实例内容，需要的朋友们可以参考下。2020-04-04
这篇文章主要介绍了Python爬虫获取基金净值信息详情，文章基于钱两篇文章的内容围绕python的相关资料展开详细介绍，需要的小伙伴可以参考一下2022-05-05
网络爬虫就是按照一定规则自动访问互联网上的信息并把内容下载下来的程序或脚本,下面这篇文章主要给大家介绍了关于如何用六步教会你使用python爬虫爬取数据的相关资料,需要的朋友可以参考下2022-04-04
这篇文章主要介绍了Django中URL视图函数的一些高级概念,Django是Python重多人气框架中最为著名的一个,需要的朋友可以参考下2015-07-07
在本篇文章里小编给大家整理的是一篇关于python爬虫判断招聘信息是否存在的实例代码，有兴趣的朋友们可以学习下。2020-11-11
这篇文章主要介绍了Django 解决新建表删除后无法重新创建等问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-05-05
最新评论