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

bugfix on canonizer

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