org.nd4j.linalg.api.activation

Interface ActivationFunction

All Superinterfaces:

com.google.common.base.Function<INDArray,INDArray>, Serializable

All Known Implementing Classes:

BaseActivationFunction, Exp, HardTanh, Linear, RectifiedLinear, RoundedLinear, Sigmoid, SoftMax, Tanh

public interface ActivationFunction
extends com.google.common.base.Function<INDArray,INDArray>, Serializable

An activation function for a hidden layer for neural networks

Author:

Adam Gibson

Method Summary

Methods

Modifier and Type	Method and Description
INDArray	applyDerivative(INDArray input) Applies the derivative of this function
Class<? extends ElementWiseOp>	transformClazz() The class used for transformation
String	type() Name of the function

Methods inherited from interface com.google.common.base.Function

apply, equals

Method Detail

transformClazz

Class<? extends ElementWiseOp> transformClazz()

The class used for transformation

Returns:

the class used for transformation

type

String type()

Name of the function

Returns:

the name of the function

applyDerivative

INDArray applyDerivative(INDArray input)

Applies the derivative of this function

Parameters:

input - the input to apply it to

Returns:

the derivative of this function applied to the input