Hi,
I have a tensor of size [24,14, 14, 30] ([Batch_size, window_H, window_W, predictions]) and the mask of size [24, 14, 14].
I want only the predictions for the windows where the mask is 1.
So, my output vector could be of size [24,14, 14, 30] or just [30].
But I am unable to index using mask.
target_tensor.shape
[24, 14, 14, 30]
contains_object_mask.shape
[24, 14, 14]
Result_tensor.shape
[24, 14, 14, 30]
or
[30]
Thank You