未验证 提交 4a4a0369 编写于 作者: A Aurelius84 提交者: GitHub

[Eager]Fix assert statement (#43492)

上级 068f48d8
......@@ -2853,8 +2853,7 @@ def tile(x, repeat_times, name=None):
"""
if in_dygraph_mode():
if isinstance(repeat_times, core.eager.Tensor):
assert (repeat_times.ndim == 1,
"Only support ndim == 1 while repeat_times is a Tensor.")
assert repeat_times.ndim == 1, "Only support ndim == 1 while repeat_times is a Tensor."
repeat_times = repeat_times.numpy().tolist()
return _C_ops.final_state_tile(x, repeat_times)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册