未验证 提交 c0e7e119 编写于 作者: H Houjiang Chen 提交者: GitHub
上级 e3810240
......@@ -147,7 +147,7 @@ def view_op(input, *shape):
if len(shape) == 1:
new_shape = shape[0]
if isinstance(new_shape, int):
new_shape = tuple(new_shape)
new_shape = (new_shape,)
else:
new_shape = shape
return flow._C.reshape(input, new_shape)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册