Skip to content

P
Paddle

机器未来 / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
提交 e76eb853 编写于 作者: C chengduoZH
浏览文件
操作

fix SwitchOrderLayer forward

上级 bc9f4c55
显示空白变更内容
内联 并排
Showing with 1 addition and 2 deletion
paddle/gserver/layers/SwitchOrderLayer.cpp
浏览文件 @ e76eb853
...@@ -83,8 +83,7 @@ void SwitchOrderLayer::forward(PassType passType) { ...@@ -83,8 +83,7 @@ void SwitchOrderLayer::forward(PassType passType) {
setOutDims(); setOutDims();
resetOutput(outDims_[0], outDims_[1] * outDims_[2] * outDims_[3]); resetOutput(outDims_[0], outDims_[1] * outDims_[2] * outDims_[3]);
if (heightAxis_.size() > 0) { if (heightAxis_.size() > 0) {
getOutputValue()->reshape(reshapeHeight_, reshapeWidth_); resetOutput(reshapeHeight_, reshapeWidth_);
getOutputGrad()->reshape(reshapeHeight_, reshapeWidth_);
} }
// switch NCHW to NHWC // switch NCHW to NHWC
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或