org.nd4j.linalg.activations

Class BaseActivationFunction

java.lang.Object

org.nd4j.linalg.activations.BaseActivationFunction

All Implemented Interfaces:

Serializable, IActivation

Direct Known Subclasses:

ActivationCube, ActivationELU, ActivationHardSigmoid, ActivationHardTanH, ActivationIdentity, ActivationLReLU, ActivationRationalTanh, ActivationRectifiedTanh, ActivationReLU, ActivationReLU6, ActivationRReLU, ActivationSELU, ActivationSigmoid, ActivationSoftmax, ActivationSoftPlus, ActivationSoftSign, ActivationSwish, ActivationTanH

public abstract class BaseActivationFunction
extends Object
implements IActivation

Base IActivation for activation functions without parameters

Author:

Alex Black

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
BaseActivationFunction()

Method Summary

Modifier and Type	Method and Description
INDArray	getGradientViewArray()
INDArray	getParametersViewArray()
int	numParams(int inputSize)
void	setGradientViewArray(INDArray viewArray)
void	setParametersViewArray(INDArray viewArray, boolean initialize)

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

backprop, getActivation

Constructor Detail

BaseActivationFunction

public BaseActivationFunction()

Method Detail

numParams

public int numParams(int inputSize)

Specified by:

numParams in interface IActivation

setParametersViewArray

public void setParametersViewArray(INDArray viewArray,
                                   boolean initialize)

Specified by:

setParametersViewArray in interface IActivation

getParametersViewArray

public INDArray getParametersViewArray()

Specified by:

getParametersViewArray in interface IActivation

setGradientViewArray

public void setGradientViewArray(INDArray viewArray)

Specified by:

setGradientViewArray in interface IActivation

getGradientViewArray

public INDArray getGradientViewArray()

Specified by:

getGradientViewArray in interface IActivation