public INDArray reduceAmean(INDArray in, INDArray dimensions, boolean keepDims) {
    NDValidation.validateNumerical("reduceAmean", "in", in);
    NDValidation.validateNumerical("reduceAmean", "dimensions", dimensions);
    return Nd4j.exec(new org.nd4j.linalg.api.ops.impl.reduce.floating.AMean(in, dimensions, keepDims));
  }