layerstring.hpp

layerstring.hpp

class TestVisitor : public boost::static_visitor<std::string>
{
 public:
  //! Set the gradient to zero for the given module.
  TestVisitor()
  {

  }

  //! Set the gradient to zero.
  template<typename LayerType>
  std::string operator()(LayerType* layer) const
  {
    return LayerString(layer);

  }

 private:
  //! Set the gradient to zero if the module implements the Gradient() function.
  template<typename T>
  typename std::enable_if<
      std::is_same<T, Linear<> >::value, std::string>::type
  LayerString(T* layer) const
  {
    return "linear";

  }

  //! Do not set the gradient to zero if the module doesn't implement the
  //! Gradient() function.
  template<typename T>
  typename std::enable_if<
      !std::is_same<T, Linear<> >::value, std::string>::type
  LayerString(T* layer) const
  {
    return "not linear";

  }
};

LayerTypes<> layerA = new Linear<>(20, 30);
LayerTypes<> layerB = new LinearNoBias<>(20, 30);



std::cout << boost::apply_visitor(TestVisitor(), layerB) << std::endl;

Thanks @zoq, I will keep in mind these include statements. I tried out your suggestion and it worked! There's just this thing that bothers me and that is the size of each function prototype because for around over 10-15 layers it would look really large and maybe slightly confusing to someone else. However, it still seems to be the best way out, so I am going with it.

I finished it a while ago and it is working fine. I'll write a test and push it :)