[pycaffe] expose Net::Reshape

diff --git a/python/caffe/_caffe.cpp b/python/caffe/_caffe.cpp
index 39e5d09..5a81a42 100644 (file)
--- a/python/caffe/_caffe.cpp
+++ b/python/caffe/_caffe.cpp
@@ -149,6 +149,7 @@ BOOST_PYTHON_MODULE(_caffe) {
       .def(bp::init<string>())
       .def("_forward",              &PyNet::Forward)
       .def("_backward",             &PyNet::Backward)
+      .def("reshape",               &PyNet::Reshape)
       .def("set_mode_cpu",          &PyNet::set_mode_cpu)
       .def("set_mode_gpu",          &PyNet::set_mode_gpu)
       .def("set_phase_train",       &PyNet::set_phase_train)

diff --git a/python/caffe/_caffe.hpp b/python/caffe/_caffe.hpp
index 5884900..ba04d27 100644 (file)
--- a/python/caffe/_caffe.hpp
+++ b/python/caffe/_caffe.hpp
@@ -98,6 +98,7 @@ class PyNet {
 
   void Forward(int start, int end) { net_->ForwardFromTo(start, end); }
   void Backward(int start, int end) { net_->BackwardFromTo(start, end); }
+  void Reshape() { net_->Reshape(); }
 
   void set_input_arrays(bp::object data_obj, bp::object labels_obj);