docs: private int[] convertBatch(int[] batch) {
          int[] conv = new int[batch.length];
          for (int i=0; i<batch.length; i++) {
              conv[i] = sample[batch[i]];
          }
          return conv;
      }