docs: public void revertFeatures(@NonNull INDArray[] features, INDArray[] maskArrays) {
          for (int i = 0; i < features.length; i++) {
              INDArray mask = (maskArrays == null ? null : maskArrays[i]);
              revertFeatures(features[i], mask, i);
          }
      }