ELU implemented

2017-06-28 11:56:43 +00:00
parent 0767df43a2
commit a86df107be
10 changed files with 123 additions and 24 deletions
@@ -8,7 +8,8 @@ Layer::Layer(Network *net, dataDim_t in_dim) {

    this->net = net;
    this->input_dim = in_dim;
-
+    this->output_dim = in_dim;
+    
    checkCUDNN( cudnnCreateTensorDescriptor(&srcTensorDesc) );
    checkCUDNN( cudnnCreateTensorDescriptor(&dstTensorDesc) );
 }