docs: public <K> JavaPairRDD<K, INDArray> feedForwardWithKey(JavaPairRDD<K, INDArray> featuresData, int batchSize) {
          return feedForwardWithMaskAndKey(featuresData.mapToPair(new SingleToPairFunction<K>()), batchSize);
      }