docs: @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;
      }