Search in sources :

Example 1 with RectifiedTanhDerivative

use of org.nd4j.linalg.api.ops.impl.transforms.gradient.RectifiedTanhDerivative in project nd4j by deeplearning4j.

the class ActivationRectifiedTanh method backprop.

@Override
public Pair<INDArray, INDArray> backprop(INDArray in, INDArray epsilon) {
    INDArray dLdz = Nd4j.getExecutioner().execAndReturn(new RectifiedTanhDerivative(in));
    dLdz.muli(epsilon);
    return new Pair<>(dLdz, null);
}
Also used : INDArray(org.nd4j.linalg.api.ndarray.INDArray) RectifiedTanhDerivative(org.nd4j.linalg.api.ops.impl.transforms.gradient.RectifiedTanhDerivative) Pair(org.nd4j.linalg.primitives.Pair)

Aggregations

INDArray (org.nd4j.linalg.api.ndarray.INDArray)1 RectifiedTanhDerivative (org.nd4j.linalg.api.ops.impl.transforms.gradient.RectifiedTanhDerivative)1 Pair (org.nd4j.linalg.primitives.Pair)1