   auto g = New<ExpressionGraph>();
   load(g, "/home/marcinj/Badania/amunmt/test2/model.npz");
 
-  size_t batchSize = 80;
+  size_t batchSize = 40;
 
   boost::timer::cpu_timer timer;
   for(int i = 1; i <= 1000; ++i) {

    joinedStates.push_back(concatenate({*itFw++, *itBw++}, 1));
  }

  // add proper axes and make this a 3D tensor
  auto encContext = name(concatenate(joinedStates, 2), "context");

  //auto decStartState = mean(encContext, axis=2);
}


    g->forward();
    //exit(0);
    //g->backward();
    if(i % 100 == 0)
      std::cout << i << std::endl;
  }
