Search in sources :

Example 1 with IAMin

use of org.nd4j.linalg.api.ops.impl.indexaccum.IAMin in project nd4j by deeplearning4j.

the class NativeOpExecutionerTest method testIMinIAMin.

@Test
public void testIMinIAMin() {
    INDArray arr = Nd4j.create(new double[] { -0.24, -0.26, -0.07, -0.01 });
    double imin = Nd4j.getExecutioner().execAndReturn(new IMin(arr.dup())).getFinalResult();
    double iamin = Nd4j.getExecutioner().execAndReturn(new IAMin(arr.dup())).getFinalResult();
    System.out.println("IMin: " + imin);
    System.out.println("IAMin: " + iamin);
    assertEquals(1, imin, 0.0);
    assertEquals(3, iamin, 0.0);
}
Also used : INDArray(org.nd4j.linalg.api.ndarray.INDArray) IAMin(org.nd4j.linalg.api.ops.impl.indexaccum.IAMin) IMin(org.nd4j.linalg.api.ops.impl.indexaccum.IMin) Test(org.junit.Test)

Example 2 with IAMin

use of org.nd4j.linalg.api.ops.impl.indexaccum.IAMin in project nd4j by deeplearning4j.

the class Nd4jTestsC method testIMinIAMin.

@Test
public void testIMinIAMin() {
    INDArray arr = Nd4j.create(new double[] { -0.24, -0.26, -0.07, -0.01 });
    INDArray abs = Transforms.abs(arr);
    IAMin iaMin = new IAMin(abs);
    IMin iMin = new IMin(arr.dup());
    double imin = Nd4j.getExecutioner().execAndReturn(iMin).getFinalResult();
    double iamin = Nd4j.getExecutioner().execAndReturn(iaMin).getFinalResult();
    System.out.println("IMin: " + imin);
    System.out.println("IAMin: " + iamin);
    assertEquals(3, iamin, 1e-12);
    assertEquals(1, imin, 1e-12);
}
Also used : INDArray(org.nd4j.linalg.api.ndarray.INDArray) IAMin(org.nd4j.linalg.api.ops.impl.indexaccum.IAMin) IMin(org.nd4j.linalg.api.ops.impl.indexaccum.IMin) Test(org.junit.Test)

Aggregations

Test (org.junit.Test)2 INDArray (org.nd4j.linalg.api.ndarray.INDArray)2 IAMin (org.nd4j.linalg.api.ops.impl.indexaccum.IAMin)2 IMin (org.nd4j.linalg.api.ops.impl.indexaccum.IMin)2