org.deeplearning4j.nn.layers.custom.testclasses

Class CustomActivation

java.lang.Object

org.nd4j.linalg.activations.BaseActivationFunction

org.deeplearning4j.nn.layers.custom.testclasses.CustomActivation

All Implemented Interfaces:

java.io.Serializable, org.nd4j.linalg.activations.IActivation

public class CustomActivation
extends org.nd4j.linalg.activations.BaseActivationFunction
implements org.nd4j.linalg.activations.IActivation

Created by Alex on 19/12/2016.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
CustomActivation()

Method Summary

Modifier and Type	Method and Description
org.apache.commons.math3.util.Pair<org.nd4j.linalg.api.ndarray.INDArray,org.nd4j.linalg.api.ndarray.INDArray>	backprop(org.nd4j.linalg.api.ndarray.INDArray in, org.nd4j.linalg.api.ndarray.INDArray epsilon)
org.nd4j.linalg.api.ndarray.INDArray	getActivation(org.nd4j.linalg.api.ndarray.INDArray in, boolean training)

Methods inherited from class org.nd4j.linalg.activations.BaseActivationFunction

getGradientViewArray, getParametersViewArray, numParams, setGradientViewArray, setParametersViewArray

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.nd4j.linalg.activations.IActivation

getGradientViewArray, getParametersViewArray, numParams, setGradientViewArray, setParametersViewArray

Constructor Detail

CustomActivation

public CustomActivation()

Method Detail

getActivation

public org.nd4j.linalg.api.ndarray.INDArray getActivation(org.nd4j.linalg.api.ndarray.INDArray in,
                                                          boolean training)

Specified by:

getActivation in interface org.nd4j.linalg.activations.IActivation

backprop

public org.apache.commons.math3.util.Pair<org.nd4j.linalg.api.ndarray.INDArray,org.nd4j.linalg.api.ndarray.INDArray> backprop(org.nd4j.linalg.api.ndarray.INDArray in,
                                                                                                                              org.nd4j.linalg.api.ndarray.INDArray epsilon)

Specified by:

backprop in interface org.nd4j.linalg.activations.IActivation