How to index a tensor using a mask and scalar

Question

I have a tensor y

import torch
y = torch.rand(1, 4, 50, 100, 2)

And I have a mask (notice, it lacks the last dimension in comparison