Examples with Pair org.nd4j.linalg.primitives.Pair used on opensource projects

Example 16 with Pair

use of org.nd4j.linalg.primitives.Pair in project nd4j by deeplearning4j.

the class ActivationSwish method backprop.

@Override
public Pair<INDArray, INDArray> backprop(INDArray in, INDArray epsilon) {
    INDArray dLdz = Nd4j.getExecutioner().execAndReturn(new SwishDerivative(in));
    dLdz.muli(epsilon);
    return new Pair<>(dLdz, null);
}

Example 17 with Pair

use of org.nd4j.linalg.primitives.Pair in project nd4j by deeplearning4j.

the class ActivationTanH method backprop.

@Override
public Pair<INDArray, INDArray> backprop(INDArray in, INDArray epsilon) {
    INDArray dLdz = Nd4j.getExecutioner().execAndReturn(new TanhDerivative(in));
    dLdz.muli(epsilon);
    return new Pair<>(dLdz, null);
}

Example 18 with Pair

use of org.nd4j.linalg.primitives.Pair in project nd4j by deeplearning4j.

the class ActivationReLU method backprop.

@Override
public Pair<INDArray, INDArray> backprop(INDArray in, INDArray epsilon) {
    INDArray dLdz = Nd4j.getExecutioner().execAndReturn(new Step(in));
    dLdz.muli(epsilon);
    return new Pair<>(dLdz, null);
}

Example 19 with Pair

use of org.nd4j.linalg.primitives.Pair in project nd4j by deeplearning4j.

the class ActivationSELU method backprop.

@Override
public Pair<INDArray, INDArray> backprop(INDArray in, INDArray epsilon) {
    INDArray dLdz = Nd4j.getExecutioner().execAndReturn(new SELUDerivative(in));
    dLdz.muli(epsilon);
    return new Pair<>(dLdz, null);
}

Example 20 with Pair

use of org.nd4j.linalg.primitives.Pair in project nd4j by deeplearning4j.

the class ActivationHardTanH method backprop.

@Override
public Pair<INDArray, INDArray> backprop(INDArray in, INDArray epsilon) {
    INDArray dLdz = Nd4j.getExecutioner().execAndReturn(new HardTanhDerivative(in));
    dLdz.muli(epsilon);
    return new Pair<>(dLdz, null);
}