今天小编就为大家分享一篇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 函数的用法详解就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持爱安网。

最新资讯
“基因魔剪”给每个癌细胞打上独特标记

“基因魔剪”给每个癌

据最新一期《科学》杂志报道,美国科学家使用CRISPR基因
细胞核起源于病毒?

细胞核起源于病毒?

细胞核的来源一直是个谜,它的起源究竟是不是如同线粒体
波场与波卡隔空互怼 太坊链上项目代币价格续刷新高

波场与波卡隔空互怼

1月25日,波场创始人孙宇晨发推表示“波场平行链拍卖”,
敲黑板!今年中国航天看什么?重点都在这里了

敲黑板!今年中国航天看

2021年,我国航天将继续再接再厉,仅中国航天科技集团就计
北斗再立新功!“极目”卫星准实时下传伽马暴观测警报

北斗再立新功!“极目”

北斗卫星导航系统再立新功!
血腥的动物竞技,是以文化之名抒发暴力欲望吗?

血腥的动物竞技,是以文

更多时候,在动物竞技背后隐藏的是非理性的暴力,有人追求
最新文章
在pycharm中为项目导入anacodna环境的操作方法

在pycharm中为项目导

这篇文章主要介绍了在pycharm中为项目导入anacodna环
tensorflow的ckpt及pb模型持久化方式及转化详解

tensorflow的ckpt及pb

今天小编就为大家分享一篇tensorflow的ckpt及pb模型持
PyTorch笔记之scatter()函数的使用

PyTorch笔记之scatter

这篇文章主要介绍了PyTorch笔记之scatter()函数的使用
python3实现网页版raspberry pi(树莓派)小车控制

python3实现网页版ras

这篇文章主要为大家详细介绍了python3实现网页版raspb
完美解决pycharm导入自己写的py文件爆红问题

完美解决pycharm导入

今天小编就为大家分享一篇完美解决pycharm导入自己写
pycharm内无法import已安装的模块问题解决

pycharm内无法import

今天小编就为大家分享一篇pycharm内无法import已安装