今天小编就为大家分享一篇Pytorch mask_select 函数的用法详解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
非常简单的函数，但是官网的介绍令人（令我）迷惑，所以稍加解释。
mask_select会将满足mask（掩码、遮罩等等，随便翻译）的指示，将满足条件的点选出来。
根据掩码张量mask中的二元值，取输入张量中的指定项( mask为一个 ByteTensor)，将取值返回到一个新的1D张量，
张量 mask须跟in来源[email protected]搞@^&代*@码网put张量有相同数量的元素数目，但形状或维度不需要相同
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 3×4]
mask = x.ge(0.5)
mask
1 1 0 1
1 1 1 0
1 0 0 0
[torch.ByteTensor of size 3×4]
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 函数的用法详解的详细内容，更多请关注gaodaima搞代码网其它相关文章！
搞代码网（gaodaima.com）提供的所有资源部分来自互联网，如果有侵犯您的版权或其他权益，请说明详细缘由并提供版权或权益证明然后发送到邮箱[email protected]，我们会在看到邮件的第一时间内为您处理，或直接联系QQ：872152909。本网站采用BY-NC-SA协议进行授权
转载请注明原文链接：Pytorch mask_select 函数的用法详解
转载请注明原文链接：Pytorch mask_select 函数的用法详解
