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爬取附近餐馆信息代码示例
Dec 09 Python
Python中的二维数组实例(list与numpy.array)
Apr 13 Python
python3.6的venv模块使用详解
Aug 01 Python
Python中创建二维数组
Oct 17 Python
对python中的six.moves模块的下载函数urlretrieve详解
Dec 19 Python
Python3实现配置文件差异对比脚本
Nov 18 Python
Python如何读取、写入CSV数据
Jul 28 Python
Python使用requests模块爬取百度翻译
Aug 25 Python
Python3 用什么IDE开发工具比较好
Nov 28 Python
python 实现端口扫描工具
Dec 18 Python
Python多线程 Queue 模块常见用法
Jul 04 Python
Python Numpy库的超详细教程
Apr 06 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
基于php的CMS中展示文章类实例分析
2015/06/18 PHP
PHP的swoole扩展安装方法详细教程
2016/05/18 PHP
PHP getName()函数讲解
2019/02/03 PHP
js 目录列举函数
2008/11/06 Javascript
js 通用订单代码
2013/12/23 Javascript
JavaScript获取网页表单提交方式的方法
2015/04/02 Javascript
js实现的后台左侧管理菜单代码
2015/09/11 Javascript
javascript中的previousSibling和nextSibling的正确用法
2015/09/16 Javascript
javascript中arguments,callee,caller详解
2016/03/16 Javascript
js根据手机客户端浏览器类型,判断跳转官网/手机网站多个实例代码
2016/04/30 Javascript
Vue.js报错Failed to resolve filter问题的解决方法
2016/05/25 Javascript
jQuery+CSS3文字跑马灯特效的简单实现
2016/06/25 Javascript
js基础之DOM中元素对象的属性方法详解
2016/10/28 Javascript
浅谈JavaScript的计时器对象
2016/12/26 Javascript
WebSocket实现简单客服聊天系统
2017/05/12 Javascript
JavaScript 上传文件(psd,压缩包等),图片,视频的实现方法
2017/06/19 Javascript
Javascript 实现 Excel 导入生成图表功能
2018/10/22 Javascript
vue鼠标悬停事件实例详解
2019/04/01 Javascript
一百行JS代码实现一个校验工具
2019/04/30 Javascript
vue项目中实现缓存的最佳方案详解
2019/07/11 Javascript
基于form-data请求格式详解
2019/10/29 Javascript
JavaScript 判断数据类型的4种方法
2020/09/11 Javascript
python读取Android permission文件
2013/11/01 Python
跟老齐学Python之画圈还不简单吗?
2014/09/20 Python
python中list列表的高级函数
2016/05/17 Python
python 接口返回的json字符串实例
2018/03/27 Python
Python import与from import使用及区别介绍
2018/09/06 Python
python得到一个excel的全部sheet标签值方法
2018/12/10 Python
对tensorflow中cifar-10文档的Read操作详解
2020/02/10 Python
Python docutils文档编译过程方法解析
2020/06/23 Python
草莓网中国:StrawberryNet中国
2020/08/17 全球购物
MYSQL支持事务吗
2013/08/09 面试题
灵山大佛导游词
2015/02/04 职场文书
法定授权委托证明书
2015/06/18 职场文书
详解Redis复制原理
2021/06/04 Redis
教你使用Python获取QQ音乐某个歌手的歌单
2022/04/03 Python