今天小编就为大家分享一篇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