提交 b66b925b 编写于 作者: P phlrain

fix concat shape check; test=develop

上级 6abac88c
...@@ -57,7 +57,7 @@ class ConcatOp : public framework::OperatorWithKernel { ...@@ -57,7 +57,7 @@ class ConcatOp : public framework::OperatorWithKernel {
"elements except the specify axis."); "elements except the specify axis.");
} else { } else {
// not check -1 with other in compile time // not check -1 with other in compile time
if (out_dims[j] != -1 && ins[i][j] != -1) { if (out_dims[j] > 0 && ins[i][j] > 0) {
PADDLE_ENFORCE_EQ(out_dims[j], ins[i][j], PADDLE_ENFORCE_EQ(out_dims[j], ins[i][j],
"Input tensors should have the same " "Input tensors should have the same "
"elements except the specify axis."); "elements except the specify axis.");
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册