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);
}
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);
}