未验证 提交 6e86fd37 编写于 作者: G GaoWei8 提交者: GitHub

fix concat dimension (#25606)

Fix the condition of concat dimension judgment.
上级 6452fd42
......@@ -51,7 +51,7 @@ static inline framework::DDim ComputeAndCheckShape(
}
} else {
bool check_shape =
is_runtime || (out_dims[j] > 0 && inputs_dims[i][j] > 0);
is_runtime || (inputs_dims[0][j] > 0 && inputs_dims[i][j] > 0);
if (check_shape) {
// check all shape in run time
PADDLE_ENFORCE_EQ(inputs_dims[0][j], inputs_dims[i][j],
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册