Pytorch mask_select 函数的用法详解
作者：漫漫冬程 发布时间：2023-11-20 22:01:40
标签：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]
来源：https://blog.csdn.net/SoftPoeter/article/details/81667810
0
投稿猜你喜欢
项目意义如果你想在支付宝蚂蚁森林收集很多能量种树，为环境绿化出一份力量，又或者是想每天称霸微信运动排行榜装逼，却不想出门走路，那么该pyth- 本文实例讲述了python实现在sqlite动态创建表的方法。分享给大家供大家参考。具体实现方法如下：import sqlite3 as d
- 垃圾分类是现代城市中越来越重要的问题，通过垃圾分类可以有效地减少环境污染和资源浪费。随着人工智能技术的发展，使用机器学习模型进行垃圾分类已经
- 前言在字典中查找某一个值的时候，若key不存在就会返回一个keyerror错误而不是一个默认值，如果想要返回一个默认值可以使用default
- Tqdm库比较常用，用于显示进度条。简单用法：from tqdm import tqdmfor i in tqdm(range(2)):
- 第一种：i=0sum=0a=0while i<102: if i>=1 and i％4==1: sum+=i eli
- FCKeditor为一开源多功能在线Web编辑器。官方网站：http://www.fckeditor.net/。相关安全文件参看：《在.ne
- 用python画图很多是根据z=f(x,y)来画图的，本博文将三个对应的坐标点输入画图：散点图：import matplotlib.pypl
- 无规矩不成方圆。编写代码也是，如果没有大概的框架，管理代码将会是一件很头疼的事。先看看笔者以前写的python脚本：如果只有一个用例，这样看
- 你搜索这个，你会发现好多都是np.zeros(5,2)，嗯都是复制的一个国外的帖子，然而没有翻译人家的话。然后你愤怒的关闭页面。这简直就是文
- 在收发快递填写地址的时候，我们会经常手动输入地址让程序智能识别，标准的地址比如，xx省xx市xx县/区xx路xx号，不过有时候也可以简单写：
- 第一种方法：用军哥的一键修改LNMP环境下MYSQL数据库密码脚本一键脚本肯定是非常方便。具体执行以下命令：wget http://soft
- decode()方法使用注册编码的编解码器的字符串进行解码。它默认为默认的字符串编码。语法以下是decode()方法的语法：st
- 数组去重复和数组排序'数组名次 Function Sort(ary,stra) KeepChecking =&n
- 工欲善其事必先利其器，估计大部分的pythoner都是从python IDE开始了python之路。有很多的python IDE工具可以用，
- 逻辑判断与逻辑语句对于─件事情正确与否（真假的判断) √ X根据判断的结果做不同的事情，就是我们的逻辑业务对于条件满足的判断语句，就是条件语
- sql 使用系统存储过程 sp_send_dbmail 发送电子邮件语法：sp_send_dbmail [ [ @profile_name
- web跨域请求1.为什么要有跨域限制举个例子：1.用户登录了自己的银行页面 http://mybank.com，http://mybank.
- 一、创建堆heapq有两种方式创建堆， 一种是使用一个空列表，然后使用heapq.heappush()函数把值加入堆中，另外一种就是使用he
- 简介使用Pandas的pivot方法可以将DF进行旋转变换，本文将会详细讲解pivot的秘密。使用Pivotpivot用来重组DF，使用指定
