Examples with BaseCondition org.nd4j.linalg.indexing.conditions.BaseCondition used on opensource projects

Example 6 with BaseCondition

use of org.nd4j.linalg.indexing.conditions.BaseCondition in project nd4j by deeplearning4j.

the class BooleanIndexing method or.

/**
 * Or over the whole ndarray given some condition
 *
 * @param n
 * @param cond
 * @return
 */
public static boolean or(final INDArray n, final Condition cond) {
    if (cond instanceof BaseCondition) {
        long val = (long) Nd4j.getExecutioner().exec(new MatchCondition(n, cond), Integer.MAX_VALUE).getDouble(0);
        if (val > 0)
            return true;
        else
            return false;
    } else {
        boolean ret = false;
        final AtomicBoolean a = new AtomicBoolean(ret);
        Shape.iterate(n, new CoordinateFunction() {

            @Override
            public void process(int[]... coord) {
                if (!a.get())
                    a.compareAndSet(false, a.get() || cond.apply(n.getDouble(coord[0])));
            }
        });
        return a.get();
    }
}

Example 7 with BaseCondition

use of org.nd4j.linalg.indexing.conditions.BaseCondition in project nd4j by deeplearning4j.

the class BooleanIndexing method or.

/**
 * Or over the whole ndarray given some condition, with respect to dimensions
 *
 * @param n    the ndarray to test
 * @param condition the condition to test against
 * @return true if all of the elements meet the specified
 * condition false otherwise
 */
public static boolean[] or(final INDArray n, final Condition condition, int... dimension) {
    if (!(condition instanceof BaseCondition))
        throw new UnsupportedOperationException("Only static Conditions are supported");
    MatchCondition op = new MatchCondition(n, condition);
    INDArray arr = Nd4j.getExecutioner().exec(op, dimension);
    boolean[] result = new boolean[arr.length()];
    for (int i = 0; i < arr.length(); i++) {
        if (arr.getDouble(i) > 0)
            result[i] = true;
        else
            result[i] = false;
    }
    return result;
}