Merge pull request #4045 from chengduoZH/fix_SwitchOrderLayer

fix SwitchOrderLayer forward
8 years ago · 0f42e5649e
parent 8631e796bb e76eb8534d
commit 0f42e5649e
1 changed files with 1 additions and 2 deletions
--- a/paddle/gserver/layers/SwitchOrderLayer.cpp
+++ b/paddle/gserver/layers/SwitchOrderLayer.cpp
@ -83,8 +83,7 @@ void SwitchOrderLayer::forward(PassType passType) {
  setOutDims();
  resetOutput(outDims_[0], outDims_[1] * outDims_[2] * outDims_[3]);
  if (heightAxis_.size() > 0) {
-    getOutputValue()->reshape(reshapeHeight_, reshapeWidth_);
-    getOutputGrad()->reshape(reshapeHeight_, reshapeWidth_);
+    resetOutput(reshapeHeight_, reshapeWidth_);
  }

  // switch NCHW to NHWC