dopt.nnet.layers.softmax source code

1 /**
2     Contains an implementation of the softmat activation function.
3     
4     Authors: Henry Gouk
5 */
6 module dopt.nnet.layers.softmax;
7 
8 import dopt.core;
9 import dopt.nnet;
10 import dopt.nnet.layers.util;
11 import dopt.online;
12 
13 ///
14 Layer softmax(Layer input)
15 {
16     import dopt.core.ops.nnet : softmax;
17     
18     return new Layer([input], input.output.softmax(), input.trainOutput.softmax, null);
19 }