Extract multiple features in a single Forward pass

I am not sure if you will need many indices - all features will come from the same image, so the index for them will be identical. So a single image_index would suffice.