Pytorch mask_select 函数的用法详解


Posted in Python onFebruary 18, 2020

非常简单的函数,但是官网的介绍令人(令我)迷惑,所以稍加解释。

mask_select会将满足mask(掩码、遮罩等等,随便翻译)的指示,将满足条件的点选出来。

根据掩码张量mask中的二元值,取输入张量中的指定项( mask为一个 ByteTensor),将取值返回到一个新的1D张量,

张量 mask须跟input张量有相同数量的元素数目,但形状或维度不需要相同

x = torch.randn(3, 4)
x
1.2045 2.4084 0.4001 1.1372
0.5596 1.5677 0.6219 -0.7954
1.3635 -1.2313 -0.5414 -1.8478
[torch.FloatTensor of size 3x4]
mask = x.ge(0.5)
mask
1 1 0 1
1 1 1 0
1 0 0 0
[torch.ByteTensor of size 3x4]
torch.masked_select(x, mask)
1.2045
2.4084
1.1372
0.5596
1.5677
0.6219
1.3635
[torch.FloatTensor of size 7]

以上这篇Pytorch mask_select 函数的用法详解就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python使用Flask框架同时上传多个文件的方法
Mar 21 Python
python&MongoDB爬取图书馆借阅记录
Feb 05 Python
Python 判断是否为质数或素数的实例
Oct 30 Python
基于Python实现的ID3决策树功能示例
Jan 02 Python
PyQt5实现QLineEdit添加clicked信号的方法
Jun 25 Python
pytorch方法测试——激活函数(ReLU)详解
Jan 15 Python
Python IDE环境之 新版Pycharm安装详细教程
Mar 05 Python
Selenium 滚动页面至元素可见的方法
Mar 18 Python
Keras设置以及获取权重的实现
Jun 19 Python
python连接mysql数据库并读取数据的实现
Sep 25 Python
解决numpy和torch数据类型转化的问题
May 23 Python
Python中基础数据类型 set集合知识点总结
Aug 02 Python
pytorch masked_fill报错的解决
Feb 18 #Python
python分别打包出32位和64位应用程序
Feb 18 #Python
pandas和spark dataframe互相转换实例详解
Feb 18 #Python
PyTorch中Tensor的数据统计示例
Feb 17 #Python
pytorch实现Tensor变量之间的转换
Feb 17 #Python
Macbook安装Python最新版本、GUI开发环境、图像处理、视频处理环境详解
Feb 17 #Python
PyCharm无法识别PyQt5的2种解决方法,ModuleNotFoundError: No module named 'pyqt5'
Feb 17 #Python
You might like
Windows PHP5和Apache的安装与配置
2009/06/08 PHP
深入理解PHP中的global
2014/08/19 PHP
详细解读php的命名空间(二)
2018/02/21 PHP
取得父标签
2006/11/14 Javascript
javascript 关闭IE6、IE7
2009/06/01 Javascript
DOM 脚本编程中的兄弟节点
2009/10/31 Javascript
js动态为代码着色显示行号
2013/05/29 Javascript
JS小游戏之象棋暗棋源码详解
2014/09/25 Javascript
JavaScript中join()方法的使用简介
2015/06/09 Javascript
jquery实现文本框textarea自适应高度
2016/03/09 Javascript
jquery $.trim()去除字符串空格的实现方法【附图例】
2016/03/30 Javascript
NodeJS中的MongoDB快速入门详细教程
2016/11/11 NodeJs
微信小程序开发经验总结(推荐)
2017/01/11 Javascript
Vue 创建组件的两种方法小结(必看)
2018/02/23 Javascript
angular 组件通信的几种实现方式
2018/07/13 Javascript
解决百度Echarts图表坐标轴越界的方法
2018/10/17 Javascript
微信小程序template模版的使用方法
2019/04/13 Javascript
浅析Git版本控制器使用
2017/12/10 Python
python文本数据相似度的度量
2018/03/12 Python
使用 Python 实现文件递归遍历的三种方式
2018/07/18 Python
Python实现DDos攻击实例详解
2019/02/02 Python
Python学习笔记之图片人脸检测识别实例教程
2019/03/06 Python
python实现共轭梯度法
2019/07/03 Python
Python基于Dlib的人脸识别系统的实现
2020/02/26 Python
Django Auth用户认证组件实现代码
2020/10/13 Python
python 如何把docker-compose.yaml导入到数据库相关条目里
2021/01/15 Python
初一体育教学反思
2014/01/29 职场文书
厨师个人自我鉴定范文
2014/04/19 职场文书
项目经理任命书
2014/06/04 职场文书
先进人物事迹材料
2014/12/29 职场文书
实习科室评语
2015/01/04 职场文书
单位证明范文
2015/06/18 职场文书
春季运动会加油词
2015/07/18 职场文书
tensorflow中的梯度求解及梯度裁剪操作
2021/05/26 Python
Golang 入门 之url 包
2022/05/04 Golang
使用CSS实现百叶窗效果示例代码
2023/05/07 HTML / CSS