提交 2c36bb4b authored 作者: Olivier Breuleux's avatar Olivier Breuleux

bugfix on canonizer

上级 f267e481
...@@ -378,6 +378,9 @@ class Canonizer(gof.LocalOptimizer): ...@@ -378,6 +378,9 @@ class Canonizer(gof.LocalOptimizer):
return self.reciprocal(self.merge_num_denum(denum, [])) return self.reciprocal(self.merge_num_denum(denum, []))
if not ld: if not ld:
if ln == 1: if ln == 1:
if isinstance(num[0], N.ndarray):
return T.as_tensor(num[0])
else:
return num[0] return num[0]
else: else:
return self.main(*num) return self.main(*num)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论