code: @Override
      public DataSet reshape(int rows, int cols) {
          DataSet ret = new DataSet(getFeatures().reshape(new long[] {rows, cols}), getLabels());
          return ret;
      }
docs: Reshapes the input in to the given rows and columns
  
  @param rows the row size
  @param cols the column size
  @return a copy of this data op with the input resized