提交 9c1cad6e 编写于 作者: T Topdu

fix windows bug of abinet

上级 bb95527f
...@@ -273,7 +273,7 @@ def _get_length(logit): ...@@ -273,7 +273,7 @@ def _get_length(logit):
out = out.cast('int32') out = out.cast('int32')
out = out.argmax(-1) out = out.argmax(-1)
out = out + 1 out = out + 1
out = paddle.where(abn, out, paddle.to_tensor(logit.shape[1])) out = paddle.where(abn, out, paddle.full_like(out, logit.shape[1]))
return out return out
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册