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

Example 11 with IMax

use of org.nd4j.linalg.api.ops.impl.indexaccum.IMax 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 12 with IMax

use of org.nd4j.linalg.api.ops.impl.indexaccum.IMax 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);
}

Example 13 with IMax

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

the class Nd4jTestsC method testIMax2Of3d.

@Test
public void testIMax2Of3d() {
    double[][][] slices = new double[3][][];
    slices[0] = new double[][] { { 1, 10, 2 }, { 3, 4, 5 } };
    slices[1] = new double[][] { { -10, -9, -8 }, { -7, -6, -5 } };
    slices[2] = new double[][] { { 4, 3, 2 }, { 1, 0, -1 } };
    // Based on a c-order traversal of each tensor
    double[] imax = new double[] { 1, 5, 0 };
    INDArray arr = Nd4j.create(3, 2, 3);
    for (int i = 0; i < 3; i++) {
        arr.get(NDArrayIndex.point(i), NDArrayIndex.all(), NDArrayIndex.all()).assign(Nd4j.create(slices[i]));
    }
    INDArray out = Nd4j.getExecutioner().exec(new IMax(arr), 1, 2);
    INDArray exp = Nd4j.create(imax);
    assertEquals(exp, out);
}

Example 14 with IMax

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

the class OpExecutionerTests method testIMax.

@Test
public void testIMax() {
    INDArray arr = Nd4j.linspace(1, 10, 10);
    IMax imax = new IMax(arr);
    assertEquals(9, ((IndexAccumulation) Nd4j.getExecutioner().exec(imax)).getFinalResult());
    arr.muli(-1);
    imax = new IMax(arr);
    int maxIdx = ((IndexAccumulation) Nd4j.getExecutioner().exec(imax)).getFinalResult();
    assertEquals(0, maxIdx);
}

Example 15 with IMax

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

the class OpExecutionerTestsC method testIMax.

@Test
public void testIMax() {
    INDArray arr = Nd4j.linspace(1, 10, 10);
    IMax imax = new IMax(arr);
    assertEquals(9, ((IndexAccumulation) Nd4j.getExecutioner().exec(imax)).getFinalResult());
    arr.muli(-1);
    imax = new IMax(arr);
    int maxIdx = ((IndexAccumulation) Nd4j.getExecutioner().exec(imax)).getFinalResult();
    assertEquals(0, maxIdx);
}