copy_visitor_impl.hpp

Go to the documentation of this file.

#ifndef MLPACK_METHODS_ANN_VISITOR_COPY_VISITOR_IMPL_HPP
#define MLPACK_METHODS_ANN_VISITOR_COPY_VISITOR_IMPL_HPP

#include <mlpack/methods/ann/layer/layer_types.hpp>
#include <boost/variant.hpp>

namespace mlpack {
namespace ann {

template <typename... CustomLayers>
template <typename LayerType>
inline LayerTypes<CustomLayers...>
CopyVisitor<CustomLayers...>::operator()(LayerType* layer) const
{
  return new LayerType(*layer);
}

template <typename... CustomLayers>
inline LayerTypes<CustomLayers...>
CopyVisitor<CustomLayers...>::operator()(MoreTypes layer) const
{
  return layer.apply_visitor(*this);
}

} // namespace ann
} // namespace mlpack

#endif