Pytorch mask_select 函数的用法详解
更新时间：2020年02月18日 08:53:27 作者：漫漫冬程
今天小编就为大家分享一篇Pytorch mask_select 函数的用法详解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
非常简单的函数，但是官网的介绍令人（令我）迷惑，所以稍加解释。
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 函数的用法详解就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要为大家介绍了如何使用pip安装清华大学镜像源步骤详解,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪2023-10-10
这篇文章主要介绍了Python logging日志库空间不足问题解决,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-09-09
今天小编就为大家分享一篇python多个模块py文件的数据共享实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-01-01
今天小编就为大家分享一篇tensorflow对图像进行拼接的例子，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-02-02
Python之Django自动实现html代码(下拉框,数据选择)
这篇文章主要介绍了Python之Django自动实现html代码(下拉框,数据选择)，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-03-03
这篇文章主要介绍了python解析多层json操作,结合实例形式分析了Python针对多层json文件的读取、解析、修改、保存等相关操作技巧,需要的朋友可以参考下2019-12-12
Windows平台Python编程必会模块之pywin32介绍
在Windows平台上，从原来使用C/C++编写原生EXE程序，到使用Python编写一些常用脚本程序，成熟的模块的使用使得编程效率大大提高了2019-10-10
这篇文章主要介绍了python实现telnet客户端的方法,分析了Python中telnetlib模块实现telnet操作的方法,并实例叙述了Telnet客户端的实现技巧,需要的朋友可以参考下2015-04-04
基于本地知识的问答机器人langchain-ChatGLM 大语言模型实现方法详解
这篇文章主要介绍了基于本地知识的问答机器人langchain-ChatGLM 大语言模型实现方法,结合具体实例形式详细分析了langchain-ChatGLM的功能、原理、部署方法与操作注意事项,需要的朋友可以参考下2023-07-07
WTForms 是用于web开发的灵活的表单验证和呈现库，它可以与您选择的任何web框架和模板引擎一起工作，并支持数据验证、CSRF保护、国际化等。本文将运用WTForms实现用户注册功能，需要的可以参考一下2022-12-12
最新评论