better network model

2017-08-01 23:03:02 +02:00
parent 300b0af5dd
commit e8355cee67
22 changed files with 166 additions and 179 deletions
@@ -5,17 +5,11 @@

 namespace tkDNN {

-Route::Route(Network *net, int *layers_id, int layers_n) : 
-    Layer(net, dataDim_t()) {
+Route::Route(Network *net, Layer **layers, int layers_n) : Layer(net) {

+    this->layers = layers;
    this->layers_n = layers_n;

-    //get layers
-    layers = new Layer*[layers_n];
-    for(int i=0; i<layers_n; i++) 
-        layers[i] = net->layers[layers_id[i]]; 
-        
-
    //get dims
    output_dim.l = 1;
    output_dim.c = 0;