org.nd4j.linalg.api.activation.MaxOut Maven / Gradle / Ivy
package org.nd4j.linalg.api.activation;
import org.nd4j.linalg.api.ndarray.INDArray;
import org.nd4j.linalg.factory.Nd4j;
import org.nd4j.linalg.ops.ElementWiseOp;
/**
* Created by agibsonccc on 10/3/14.
*/
public class MaxOut extends BaseActivationFunction {
/**
* The class used for transformation
*
* @return the class used for transformation
*/
@Override
public Class transformClazz() {
return null;
}
/**
* Applies the derivative of this function
*
* @param input the input to apply it to
* @return the derivative of this function applied to
* the input
*/
@Override
public INDArray applyDerivative(INDArray input) {
return Nd4j.ones(input.shape());
}
}