使用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 运算符 供重载参考
Jun 11 Python
详解Python中映射类型的内建函数和工厂函数
Aug 19 Python
Python之父谈Python的未来形式
Jul 01 Python
一篇文章搞懂Python的类与对象名称空间
Dec 10 Python
使用python实现抓取腾讯视频所有电影的爬虫
Apr 15 Python
解析python的局部变量和全局变量
Aug 15 Python
Python生态圈图像格式转换问题(推荐)
Dec 02 Python
AUC计算方法与Python实现代码
Feb 28 Python
python计算导数并绘图的实例
Feb 29 Python
python实现音乐播放和下载小程序功能
Apr 26 Python
Python 可视化神器Plotly详解
Dec 26 Python
用Python提取PDF表格的方法
Apr 11 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
一个ORACLE分页程序,挺实用的.
2006/10/09 PHP
探讨如何使用SimpleXML函数来加载和解析XML文档
2013/06/07 PHP
PHP自动重命名文件实现方法
2014/11/04 PHP
Thinkphp微信公众号支付接口
2016/08/04 PHP
Yii2框架制作RESTful风格的API快速入门教程
2016/11/08 PHP
在Yii2特定页面如何禁用调试工具栏Debug Toolbar详解
2017/08/07 PHP
Laravel Reponse响应客户端示例详解
2020/09/03 PHP
javascript div 遮罩层封锁整个页面
2009/07/10 Javascript
javascript 运算数的求值顺序
2011/08/23 Javascript
javascript管中窥豹 形参与实参浅析
2011/12/17 Javascript
文件编码导致jquery失效的解决方法
2013/06/26 Javascript
js jquery获取随机生成id的服务器控件的三种方法
2013/07/11 Javascript
关于img的href和src取变量及赋值的方法
2014/04/28 Javascript
解决js下referer兼容各大浏览器的方法
2014/11/03 Javascript
jQuery中的编程范式详解
2014/12/15 Javascript
javascript学习笔记整理(概述、变量、数据类型简介)
2015/10/25 Javascript
JavaScript原生数组Array常用方法
2017/04/06 Javascript
js实现图片轮播效果学习笔记
2017/07/26 Javascript
解决vue js IOS H5focus无法自动弹出键盘的问题
2018/08/30 Javascript
vue-cli 构建骨架屏的方法示例
2018/11/08 Javascript
python subprocess pipe 实时输出日志的操作
2020/12/05 Python
学生学习总结的自我评价
2013/10/22 职场文书
个人收入证明范本
2014/01/12 职场文书
乡镇保密工作责任书
2014/07/28 职场文书
公司领导班子群众路线四风问题对照检查材料
2014/10/02 职场文书
2014年中学生检讨书大全
2014/10/09 职场文书
师德师风自查材料
2014/10/14 职场文书
三峡人家导游词
2015/01/31 职场文书
2015年监理工作总结范文
2015/04/07 职场文书
法律进社区活动总结
2015/05/07 职场文书
第二次离婚起诉书
2015/05/18 职场文书
帝企鹅日记观后感
2015/06/10 职场文书
开业庆典嘉宾致辞
2015/08/01 职场文书
财务管理制度范本
2015/08/04 职场文书
2019年中学生的思想品德评语集锦
2019/12/19 职场文书
面试分析分布式架构Redis热点key大Value解决方案
2022/03/13 Redis