提交 8bb548e8 authored 作者: Frederic Bastien's avatar Frederic Bastien

return a list of tuple in the Elemwise.infer_shape method to me more constant.

上级 4b6d0d75
...@@ -643,7 +643,7 @@ class Elemwise(Op): ...@@ -643,7 +643,7 @@ class Elemwise(Op):
break break
assert b_dim, 'BB' assert b_dim, 'BB'
oshp.append(b_dim) oshp.append(b_dim)
rval.append(oshp) rval.append(tuple(oshp))
return rval return rval
def _c_all(self, node, name, inames, onames, sub): def _c_all(self, node, name, inames, onames, sub):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论