org.nd4j.linalg.api.activation.ActivationFunction Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of nd4j-api Show documentation

There is a newer version: 1.0.0-M2.1

package org.nd4j.linalg.api.activation;

import com.google.common.base.Function;
import org.nd4j.linalg.api.ndarray.INDArray;
import org.nd4j.linalg.ops.ElementWiseOp;


import java.io.Serializable;

/**
 * An activation function for a hidden layer for neural networks
 * @author Adam Gibson
 *
 */
public interface ActivationFunction extends Function,Serializable {


    /**
     * The class used for transformation
     * @return the class used for transformation
     */
    public Class transformClazz();

    /**
     * Name of the function
     * @return the name of the function
     */
    public String type();


	/**
	 * Applies the derivative of this function
	 * @param input the input to apply it to
	 * @return the derivative of this function applied to 
	 * the input
	 */
	public INDArray applyDerivative(INDArray input);
}