org.nd4j.autodiff.validation

Class OpTestCase

java.lang.Object

org.nd4j.autodiff.validation.OpTestCase

public class OpTestCase
extends Object

Constructor Summary

Constructors

Constructor and Description
OpTestCase(@NonNull DynamicCustomOp op)

Method Summary

Modifier and Type	Method and Description
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	expectedOutput(int outputNum, @NonNull LongShapeDescriptor expShape, @NonNull Function<INDArray,String> validationFn)
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