Package org.nd4j.autodiff.validation

Class OpTestCase

java.lang.Object

org.nd4j.autodiff.validation.OpTestCase

public class OpTestCase
extends Object

Constructor Summary

Constructors

Constructor	Description
OpTestCase(@NonNull DynamicCustomOp op)

Method Summary

Modifier and Type	Method	Description
OpTestCase	expectedOutput(int outputNum, @NonNull LongShapeDescriptor expShape, @NonNull Function<INDArray,String> validationFn)
OpTestCase	expectedOutput(int outputNum, INDArray expected)	Validate the op output using INDArray.equals(INDArray)
OpTestCase	expectedOutput(int outputNum, INDArray expected, double eps)	Validate the op output using INDArray.equals(INDArray)
OpTestCase	expectedOutputRelError(int outputNum, @NonNull INDArray expected, double maxRelError, double minAbsError)	Validate the output for a single variable using element-wise relative error: relError = abs(x-y)/(abs(x)+abs(y)), with x=y=0 case defined to be 0.0.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

OpTestCase

public OpTestCase(@NonNull
                  @NonNull DynamicCustomOp op)

Method Detail

expectedOutput

public OpTestCase expectedOutput(int outputNum,
                                 INDArray expected)

Validate the op output using INDArray.equals(INDArray)

Parameters:

outputNum - Number of the output

expected - Expected INDArray

expectedOutput

public OpTestCase expectedOutput(int outputNum,
                                 INDArray expected,
                                 double eps)

Validate the op output using INDArray.equals(INDArray)

Parameters:

outputNum - Number of the output

expected - Expected INDArray

expectedOutputRelError

public OpTestCase expectedOutputRelError(int outputNum,
                                         @NonNull
                                         @NonNull INDArray expected,
                                         double maxRelError,
                                         double minAbsError)

Validate the output for a single variable using element-wise relative error: relError = abs(x-y)/(abs(x)+abs(y)), with x=y=0 case defined to be 0.0. Also has a minimum absolute error condition, which must be satisfied for the relative error failure to be considered legitimate

Parameters:

outputNum - output number

expected - Expected INDArray

maxRelError - Maximum allowable relative error

minAbsError - Minimum absolute error for a failure to be considered legitimate

expectedOutput

public OpTestCase expectedOutput(int outputNum,
                                 @NonNull
                                 @NonNull LongShapeDescriptor expShape,
                                 @NonNull
                                 @NonNull Function<INDArray,String> validationFn)

Parameters:

outputNum - Output number to check

expShape - Expected shape for the output

validationFn - Function to use to validate the correctness of the specific Op. Should return null if validation passes, or an error message if the op validation fails