How to filter out elements across a particular dimension

Question

Let\'s say tensor a has shape (128, 20, 10). I\'d like to filter this tensor into tensor b of shape (128,19,10) based on a condition: In e