public SDVariable countNonZero(String name, SDVariable in, boolean keepDims, long... dimensions) {
    SDValidation.validateNumerical("countNonZero", "in", in);
    Preconditions.checkArgument(dimensions.length >= 0, "dimensions has incorrect size/length. Expected: dimensions.length >= 0, got %s", dimensions.length);
    SDVariable out =  new org.nd4j.linalg.api.ops.impl.reduce.longer.CountNonZero(sd,in, keepDims, dimensions).outputVariable();
    return sd.updateVariableNameAndReference(out, name);