Examples with IAMax org.nd4j.linalg.api.ops.impl.indexaccum.IAMax used on opensource projects

Example 1 with IAMax

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

the class SporadicTests method testIAMax1.

@Test
public void testIAMax1() throws Exception {
    INDArray arrayX = Nd4j.rand('c', 128000, 4);
    Nd4j.getExecutioner().exec(new IAMax(arrayX), 1);
    long time1 = System.nanoTime();
    for (int i = 0; i < 10000; i++) {
        Nd4j.getExecutioner().exec(new IAMax(arrayX), 1);
    }
    long time2 = System.nanoTime();
    System.out.println("Time: " + ((time2 - time1) / 10000));
}

Example 2 with IAMax

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

the class OpExecutionerTests method testIamax2.

@Test
public void testIamax2() {
    INDArray linspace = Nd4j.linspace(1, 4, 4);
    assertEquals(getFailureMessage(), 3, Nd4j.getBlasWrapper().iamax(linspace));
    val op = new IAMax(linspace);
    int iamax = Nd4j.getExecutioner().execAndReturn(op).getFinalResult();
    assertEquals(3, iamax);
}

Example 3 with IAMax

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

the class NativeOpExecutionerTest method testIMaxIAMax.

@Test
public void testIMaxIAMax() {
    INDArray arr = Nd4j.create(new double[] { -0.24, -0.26, -0.07, -0.01 });
    double imax = Nd4j.getExecutioner().execAndReturn(new IMax(arr.dup())).getFinalResult();
    double iamax = Nd4j.getExecutioner().execAndReturn(new IAMax(arr.dup())).getFinalResult();
    System.out.println("IMAX: " + imax);
    System.out.println("IAMAX: " + iamax);
    assertEquals(3, imax, 0.0);
    assertEquals(1, iamax, 0.0);
}

Example 4 with IAMax

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

the class Nd4jTestsC method testIMaxIAMax.

@Test
public void testIMaxIAMax() {
    Nd4j.getExecutioner().setProfilingMode(OpExecutioner.ProfilingMode.ALL);
    INDArray arr = Nd4j.create(new double[] { -0.24, -0.26, -0.07, -0.01 });
    IMax iMax = new IMax(arr.dup());
    IAMax iaMax = new IAMax(arr.dup());
    double imax = Nd4j.getExecutioner().execAndReturn(iMax).getFinalResult();
    double iamax = Nd4j.getExecutioner().execAndReturn(iaMax).getFinalResult();
    System.out.println("IMAX: " + imax);
    System.out.println("IAMAX: " + iamax);
    assertEquals(1, iamax, 0.0);
    assertEquals(3, imax, 0.0);
}