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

Example 6 with IMin

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

the class OpExecutionerTestsC 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);
}