Merge pull request #4045 from chengduoZH/fix_SwitchOrderLayer

fix SwitchOrderLayer forward

Merge pull request #4045 from chengduoZH/fix_SwitchOrderLayer
fix SwitchOrderLayer forward
0f42e564 · whs · GitHub · 8631e796 · e76eb853 · 0f42e564
隐藏空白更改
内联并排

Showing with 1 addition and 2 deletion

paddle/gserver/layers/SwitchOrderLayer.cpp paddle/gserver/layers/SwitchOrderLayer.cpp +1 -2

未找到文件。
--- 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