Search in sources :

Example 6 with Op

use of org.nd4j.linalg.api.ops.Op in project nd4j by deeplearning4j.

the class Nd4jTestsC method testMatchTransform.

@Test
public void testMatchTransform() throws Exception {
    val array = Nd4j.create(new double[] { 1, 1, 1, 0, 1, 1 }, 'c');
    val exp = Nd4j.create(new double[] { 0, 0, 0, 1, 0, 0 }, 'c');
    Op op = new MatchConditionTransform(array, array, 1e-5, Conditions.epsEquals(0.0));
    Nd4j.getExecutioner().exec(op);
    assertEquals(exp, array);
}
Also used : lombok.val(lombok.val) Op(org.nd4j.linalg.api.ops.Op) BroadcastOp(org.nd4j.linalg.api.ops.BroadcastOp) DynamicCustomOp(org.nd4j.linalg.api.ops.DynamicCustomOp) Test(org.junit.Test)

Aggregations

Op (org.nd4j.linalg.api.ops.Op)6 INDArray (org.nd4j.linalg.api.ndarray.INDArray)4 Test (org.junit.Test)3 DifferentialFunction (org.nd4j.autodiff.functions.DifferentialFunction)2 BroadcastOp (org.nd4j.linalg.api.ops.BroadcastOp)2 DynamicCustomOp (org.nd4j.linalg.api.ops.DynamicCustomOp)2 CompareAndSet (org.nd4j.linalg.api.ops.impl.transforms.comparison.CompareAndSet)2 Condition (org.nd4j.linalg.indexing.conditions.Condition)2 lombok.val (lombok.val)1 NoOpNameFoundException (org.nd4j.imports.NoOpNameFoundException)1