Extract multiple features in a single Forward pass

I see. I guess the numbers are still a little redundant (as after each loop they will be the same) but the current solution seems fine to me.