org.nd4j.linalg.indexing.conditions.AbsValueGreaterThan Maven / Gradle / Ivy
package org.nd4j.linalg.indexing.conditions;
import org.apache.commons.math3.util.FastMath;
import org.nd4j.linalg.api.complex.IComplexNumber;
/**Boolean condition on absolute value: abs(x) > value
*/
public class AbsValueGreaterThan extends BaseCondition {
/**
* Special constructor for pairwise boolean operations.
*/
public AbsValueGreaterThan() {
super(0.0);
}
public AbsValueGreaterThan(Number value) {
super(value);
}
/**
* Returns condition ID for native side
*
* @return
*/
@Override
public int condtionNum() {
return 7;
}
@Override
public Boolean apply(Number input) {
return FastMath.abs(input.doubleValue()) > value.doubleValue();
}
@Override
public Boolean apply(IComplexNumber input) {
return input.absoluteValue().doubleValue() > value.doubleValue();
}
}