Efficient masking with bool for outer product of arrays

Question

I am working on a piece of code that needs to keep track of a set of indices for a later selection from a test_array, to perform a product operation. The idea is to perform