public INDArray reduceAMax(INDArray in, INDArray dimensions) {
    NDValidation.validateNumerical("reduceAMax", "in", in);
    NDValidation.validateNumerical("reduceAMax", "dimensions", dimensions);
    return Nd4j.exec(new org.nd4j.linalg.api.ops.impl.reduce.same.AMax(in, dimensions, false));
  }