Extract multiple features in a single Forward pass

The inner loop has to use multiple indices to independently log the progress of each feature.
```
    for (int i = 0; i < num_features; ++i) {
      for (int n = 0; n < batch_size; ++n) {
```