使用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中关于tqdm的用法
这篇文章主要介绍了python中关于tqdm的用法及说明,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
2023-08-08
如何基于python测量代码运行时间
这篇文章主要介绍了如何基于python测量代码运行时间,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
2019-12-12
Python编程基础之输入与输出
这篇文章主要为大家介绍了Python输入与输出，具有一定的参考价值，感兴趣的小伙伴们可以参考一下，希望能够给你带来帮助
2021-12-12
python如何实现视频转代码视频
这篇文章主要为大家详细介绍了python如何实现视频转代码视频，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2019-06-06
用Python可视化新冠疫情数据
大家好，本篇文章主要讲的是用Python可视化新冠疫情数据，感兴趣的同学赶快来看一看吧，对你有帮助的话记得收藏一下
2022-01-01
Python网络编程之TCP套接字简单用法示例
这篇文章主要介绍了Python网络编程之TCP套接字简单用法,结合实例形式分析了TCP套接字的功能及客户端、服务器端具体实现方法,需要的朋友可以参考下
2018-04-04
Python中subplots_adjust函数的用法
这篇文章主要介绍了Python中subplots_adjust函数的用法及说明,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
2023-08-08
python自动化之如何利用allure生成测试报告
这篇文章主要给大家介绍了关于python自动化之如何利用allure生成测试报告的相关资料，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2021-05-05
python 使用elasticsearch 实现翻页的三种方式
这篇文章主要介绍了python 使用elasticsearch 实现翻页的三种方式，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2020-07-07
基于本地知识的问答机器人langchain-ChatGLM 大语言模型实现方法详解
这篇文章主要介绍了基于本地知识的问答机器人langchain-ChatGLM 大语言模型实现方法,结合具体实例形式详细分析了langchain-ChatGLM的功能、原理、部署方法与操作注意事项,需要的朋友可以参考下
2023-07-07