From 64bf752dcc107bcab054b8b93b845ee4431cb103 Mon Sep 17 00:00:00 2001 From: phlrain Date: Mon, 15 Apr 2019 08:45:18 +0000 Subject: [PATCH] fix concat; test=develop --- paddle/fluid/operators/concat_op.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddle/fluid/operators/concat_op.cc b/paddle/fluid/operators/concat_op.cc index 3ba3e5a9429..b1a6d66b80e 100644 --- a/paddle/fluid/operators/concat_op.cc +++ b/paddle/fluid/operators/concat_op.cc @@ -52,7 +52,7 @@ class ConcatOp : public framework::OperatorWithKernel { if (ctx->IsRuntime()) { out_dims[axis] += ins[i][j]; } else { - if (out_dims[axis] == -1 || ins[i][j] == -1) { + if (ins[i][j] == -1) { out_dims[axis] = -1; } else { out_dims[axis] += ins[i][j]; -- GitLab