org.nd4j.linalg.indexing.conditions.AbsValueLessOrEqualsThan 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
*
* @author [email protected]
*/
public class AbsValueLessOrEqualsThan extends BaseCondition {
/**
* Special constructor for pairwise boolean operations.
*/
public AbsValueLessOrEqualsThan() {
super(0.0);
}
public AbsValueLessOrEqualsThan(Number value) {
super(value);
}
/**
* Returns condition ID for native side
*
* @return
*/
@Override
public int condtionNum() {
return 13;
}
@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();
}
}