提交 64bf752d 编写于 作者: P phlrain

fix concat; test=develop

上级 db0518bb
...@@ -52,7 +52,7 @@ class ConcatOp : public framework::OperatorWithKernel { ...@@ -52,7 +52,7 @@ class ConcatOp : public framework::OperatorWithKernel {
if (ctx->IsRuntime()) { if (ctx->IsRuntime()) {
out_dims[axis] += ins[i][j]; out_dims[axis] += ins[i][j];
} else { } else {
if (out_dims[axis] == -1 || ins[i][j] == -1) { if (ins[i][j] == -1) {
out_dims[axis] = -1; out_dims[axis] = -1;
} else { } else {
out_dims[axis] += ins[i][j]; out_dims[axis] += ins[i][j];
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册