Extract multiple features in a single Forward pass