Hi,
If you pass a bool tensor, it is interpretet as a mask and will return the entries where True is given.
Isn’t it interpretting the list of bools as a list of int with False=0 and True=1? Wrapping this in a torch.ByteTensor() will recover the mask behavior.