ELU implemented

2017-06-28 11:56:43 +00:00
parent 0767df43a2
commit a86df107be
10 changed files with 123 additions and 24 deletions
@@ -3,10 +3,22 @@

 int main() {

-    tkDNN::dataDim_t dim(1, 1, 10, 10);
-    dim.print();
    tkDNN::Network net;
-    tkDNN::Dense d(&net, dim, 2, "ci", "lol");
+    tkDNN::dataDim_t dim(1, 8, 1, 1);
+    tkDNN::Dense      d(&net, dim, 2, "../tests/dense.bin", "../tests/dense.bias.bin");
+    tkDNN::Activation a(&net, d.output_dim, tkDNN::ACTIVATION_ELU);
+
+    value_type *data;
+    value_type *input_h;
+    readBinaryFile("../tests/input.bin", 8, &input_h, &data);
+
+    dim.print();
+    data = d.infer(dim, data);
+    dim.print();
+    data = a.infer(dim, data);
+    dim.print();
+
+    printDeviceVector(dim.tot(), data);

    return 0;
 }