public interface IActivation extends Serializable
Modifier and Type | Method and Description |
---|---|
org.nd4j.linalg.primitives.Pair<INDArray,INDArray> |
backprop(INDArray in,
INDArray epsilon)
Backpropagate the errors through the activation function, given input z and epsilon dL/da.
Returns 2 INDArrays: (a) The gradient dL/dz, calculated from dL/da, and (b) The parameter gradients dL/dw, where w is the weights in the activation function. |
INDArray |
getActivation(INDArray in,
boolean training)
Carry out activation function on the input array (usually known as 'preOut' or 'z')
Implementations must overwrite "in", transform in place and return "in"
Can support separate behaviour during test
|
INDArray |
getGradientViewArray() |
INDArray |
getParametersViewArray() |
int |
numParams(int inputSize) |
void |
setGradientViewArray(INDArray viewArray) |
void |
setParametersViewArray(INDArray viewArray,
boolean initialize) |
INDArray getActivation(INDArray in, boolean training)
in
- training
- org.nd4j.linalg.primitives.Pair<INDArray,INDArray> backprop(INDArray in, INDArray epsilon)
in
- Input, before applying the activation function (z, or 'preOut')epsilon
- Gradient to be backpropagated: dL/da, where L is the loss functionint numParams(int inputSize)
void setParametersViewArray(INDArray viewArray, boolean initialize)
INDArray getParametersViewArray()
void setGradientViewArray(INDArray viewArray)
INDArray getGradientViewArray()
Copyright © 2017. All rights reserved.