Search in sources :

Example 6 with LinearActivationLayer

use of com.simiacryptus.mindseye.layers.java.LinearActivationLayer in project MindsEye by SimiaCryptus.

the class SquareActivationLayerTest method getReferenceLayer.

@Nullable
@Override
public Layer getReferenceLayer() {
    PipelineNetwork network = new PipelineNetwork();
    network.wrap(new LinearActivationLayer().setScale(alpha), network.wrap(new NthPowerActivationLayer().setPower(2), network.getInput(0)));
    return network;
// return new NthPowerActivationLayer().setPower(2);
}
Also used : NthPowerActivationLayer(com.simiacryptus.mindseye.layers.java.NthPowerActivationLayer) PipelineNetwork(com.simiacryptus.mindseye.network.PipelineNetwork) LinearActivationLayer(com.simiacryptus.mindseye.layers.java.LinearActivationLayer) Nullable(javax.annotation.Nullable)

Aggregations

LinearActivationLayer (com.simiacryptus.mindseye.layers.java.LinearActivationLayer)6 PipelineNetwork (com.simiacryptus.mindseye.network.PipelineNetwork)5 Nonnull (javax.annotation.Nonnull)4 ActivationLayer (com.simiacryptus.mindseye.layers.cudnn.ActivationLayer)3 ArrayTrainable (com.simiacryptus.mindseye.eval.ArrayTrainable)2 Trainable (com.simiacryptus.mindseye.eval.Trainable)2 Tensor (com.simiacryptus.mindseye.lang.Tensor)2 SumInputsLayer (com.simiacryptus.mindseye.layers.java.SumInputsLayer)2 IterativeTrainer (com.simiacryptus.mindseye.opt.IterativeTrainer)2 ArmijoWolfeSearch (com.simiacryptus.mindseye.opt.line.ArmijoWolfeSearch)2 QQN (com.simiacryptus.mindseye.opt.orient.QQN)2 StepRecord (com.simiacryptus.mindseye.test.StepRecord)2 ArrayList (java.util.ArrayList)2 Nullable (javax.annotation.Nullable)2 SquareActivationLayer (com.simiacryptus.mindseye.layers.cudnn.SquareActivationLayer)1 NthPowerActivationLayer (com.simiacryptus.mindseye.layers.java.NthPowerActivationLayer)1 ProductInputsLayer (com.simiacryptus.mindseye.layers.java.ProductInputsLayer)1 SigmoidActivationLayer (com.simiacryptus.mindseye.layers.java.SigmoidActivationLayer)1 DAGNode (com.simiacryptus.mindseye.network.DAGNode)1