public INDArray reduceAMax(INDArray in, INDArray dimensions, boolean keepDims) {
    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, keepDims));
  }