org.nd4j.linalg.activations

Class BaseActivationFunction

java.lang.Object

org.nd4j.linalg.activations.BaseActivationFunction

All Implemented Interfaces:

Serializable, IActivation

Direct Known Subclasses:

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

public abstract class BaseActivationFunction
extends Object
implements IActivation

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
BaseActivationFunction()

Method Summary

Modifier and Type	Method and Description
protected void	assertShape(INDArray in, INDArray epsilon)
int	numParams(int inputSize)

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

assertShape

protected void assertShape(INDArray in,
                           INDArray epsilon)