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