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

Example 1 with IMin

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

the class OpExecutionerTests method testIMin.

@Test
public void testIMin() {
    INDArray arr = Nd4j.linspace(1, 10, 10);
    IMin imin = new IMin(arr);
    assertEquals(0, ((IndexAccumulation) Nd4j.getExecutioner().exec(imin)).getFinalResult());
    arr.muli(-1);
    imin = new IMin(arr);
    int minIdx = ((IndexAccumulation) Nd4j.getExecutioner().exec(imin)).getFinalResult();
    assertEquals(9, minIdx);
}

Example 2 with IMin

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

the class CudaIndexReduceTests method testPinnedIMin.

@Test
public void testPinnedIMin() throws Exception {
    // simple way to stop test if we're not on CUDA backend here
    assertEquals("JcublasLevel1", Nd4j.getBlasWrapper().level1().getClass().getSimpleName());
    INDArray array1 = Nd4j.create(new float[] { 1.0f, 0.1f, 2.0f, 3.0f, 4.0f, 5.0f });
    int idx = ((IndexAccumulation) Nd4j.getExecutioner().exec(new IMin(array1))).getFinalResult();
    System.out.println("Array1: " + array1);
    assertEquals(1, idx);
}

Example 3 with IMin

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

Example 4 with IMin

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

Example 5 with IMin

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

the class CudaIndexReduceTests method testPinnedIMin2.

@Test
public void testPinnedIMin2() throws Exception {
    // simple way to stop test if we're not on CUDA backend here
    assertEquals("JcublasLevel1", Nd4j.getBlasWrapper().level1().getClass().getSimpleName());
    INDArray array1 = Nd4j.create(new float[] { 0.1f, 1.1f, 2.0f, 3.0f, 4.0f, 5.0f });
    int idx = ((IndexAccumulation) Nd4j.getExecutioner().exec(new IMin(array1))).getFinalResult();
    System.out.println("Array1: " + array1);
    assertEquals(0, idx);
}