code: @Override
      public int[] predict(INDArray input) {
          INDArray output = activate(input, false, LayerWorkspaceMgr.noWorkspacesImmutable());
          int[] ret = new int[input.rows()];
          for (int i = 0; i < ret.length; i++)
              ret[i] = Nd4j.getBlasWrapper().iamax(output.getRow(i));
          return ret;
      }
docs: Returns the predictions for each example in the dataset
  @param input the matrix to predict
  @return the prediction for the dataset