better network model

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

 namespace tkDNN {

-Network::Network() {
+Network::Network(dataDim_t input_dim) {
+    this->input_dim = input_dim;

    float tk_ver = float(tkDNN::getVersion())/1000;
    float cu_ver = float(cudnnGetVersion())/1000;
@@ -47,4 +48,12 @@ bool Network::addLayer(Layer *l) {
    return true;
 }

+dataDim_t Network::getOutputDim() {
+
+        if(num_layers == 0)
+            return input_dim;
+        else
+            return layers[num_layers-1]->output_dim;
+}
+
 }