使用pytorch 筛选出一定范围的值


Posted in Python onJune 28, 2020

我就废话不多说了,大家还是直接看代码吧~

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 相关文章推荐
利用Python实现Windows下的鼠标键盘模拟的实例代码
Jul 13 Python
python实现简易版计算器
Jun 22 Python
解决pycharm界面不能显示中文的问题
May 23 Python
Python3.6基于正则实现的计算器示例【无优化简单注释版】
Jun 14 Python
python实现随机漫步算法
Aug 27 Python
详解Django的model查询操作与查询性能优化
Oct 16 Python
python交易记录整合交易类详解
Jul 03 Python
Python logging设置和logger解析
Aug 28 Python
python 对一幅灰度图像进行直方图均衡化
Oct 27 Python
Python Spyder 调出缩进对齐线的操作
Feb 26 Python
Python中json.dumps()函数的使用解析
May 17 Python
Python OpenCV超详细讲解读取图像视频和网络摄像头
Apr 02 Python
解析python 中/ 和 % 和 //(地板除)
Jun 28 #Python
pytorch 常用函数 max ,eq说明
Jun 28 #Python
浅谈pytorch中torch.max和F.softmax函数的维度解释
Jun 28 #Python
Python turtle库的画笔控制说明
Jun 28 #Python
使用python修改文件并立即写回到原始位置操作(inplace读写)
Jun 28 #Python
python删除指定列或多列单个或多个内容实例
Jun 28 #Python
Python3实现建造者模式的示例代码
Jun 28 #Python
You might like
php中获取关键词及所属来源搜索引擎名称的代码
2011/02/15 PHP
php后台多用户权限组思路与实现程序代码分享
2012/02/13 PHP
phpmyadmin打开很慢的解决方法
2014/04/21 PHP
ThinkPHP查询语句与关联查询用法实例
2014/11/01 PHP
PHP怎样用正则抓取页面中的网址
2016/08/09 PHP
php处理多图上传压缩代码功能
2018/06/13 PHP
jQuery Ajax之load()方法
2009/10/12 Javascript
基于Jquery与WebMethod投票功能实现代码
2011/01/19 Javascript
javascript操作referer详细解析
2014/03/10 Javascript
Firefox下无法正常显示年份的解决方法
2014/09/04 Javascript
浅谈Node.js:理解stream
2016/12/08 Javascript
vuejs使用axios异步访问时用get和post的实例讲解
2018/08/09 Javascript
详解CommonJS和ES6模块循环加载处理的区别
2018/12/26 Javascript
sortable+element 实现表格行拖拽的方法示例
2019/06/07 Javascript
修改vue源码实现动态路由缓存的方法
2020/01/21 Javascript
javascript操作向表格中动态加载数据
2020/08/27 Javascript
vue如何使用rem适配
2021/02/06 Vue.js
python 实现堆排序算法代码
2012/06/05 Python
Python 查找list中的某个元素的所有的下标方法
2018/06/27 Python
一看就懂得Python的math模块
2018/10/21 Python
详解pytorch 0.4.0迁移指南
2019/06/16 Python
djano一对一、多对多、分页实例代码
2019/08/16 Python
将pytorch转成longtensor的简单方法
2020/02/18 Python
PySide2出现“ImportError: DLL load failed: 找不到指定的模块”的问题及解决方法
2020/06/10 Python
详解Html5 Canvas画线有毛边解决方法
2018/03/01 HTML / CSS
La Redoute英国官网:法国时尚品牌
2017/04/27 全球购物
在印度上传处方,在线订购药品:Medlife
2019/03/28 全球购物
室内设计自我鉴定
2013/10/15 职场文书
网络管理专业求职信
2014/03/15 职场文书
学校做一个有道德的人活动方案
2014/08/23 职场文书
2014年电话客服工作总结
2014/12/09 职场文书
篮球比赛通讯稿
2015/07/18 职场文书
奖学金申请个人主要事迹材料
2015/11/04 职场文书
详解gantt甘特图可拖拽、编辑(vue、react都可用 highcharts)
2021/11/27 Vue.js
TypeScript 使用 Tuple Union 声明函数重载
2022/04/07 Javascript
MySQL去除密码登录告警的方法
2022/04/20 MySQL