public INDArray[] moments(INDArray input, INDArray axes, boolean keepDims) {
    NDValidation.validateNumerical("moments", "input", input);
    NDValidation.validateNumerical("moments", "axes", axes);
    return Nd4j.exec(new org.nd4j.linalg.api.ops.impl.reduce.Moments(input, axes, keepDims));
  }