Hi,
Did you find out how to do this? I am having the same issue and I am pondering what the best method is (e.g. using masked_select and then view, or converting the boolean mask to int and doing index_select)