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

bugfix on canonizer

上级 f267e481
...@@ -378,7 +378,10 @@ class Canonizer(gof.LocalOptimizer): ...@@ -378,7 +378,10 @@ 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:
return num[0] if isinstance(num[0], N.ndarray):
return T.as_tensor(num[0])
else:
return num[0]
else: else:
return self.main(*num) return self.main(*num)
return self.inverse(self.merge_num_denum(num, []), return self.inverse(self.merge_num_denum(num, []),
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论