提交 c2b2b641 authored 作者: Frederic Bastien's avatar Frederic Bastien

better string representation.

上级 9a7d1171
...@@ -356,7 +356,7 @@ class FromFunctionLocalOptimizer(LocalOptimizer): ...@@ -356,7 +356,7 @@ class FromFunctionLocalOptimizer(LocalOptimizer):
def tracks(self): def tracks(self):
return self._tracks return self._tracks
def __str__(self): def __str__(self):
return getattr(self, 'name', '<FromFunctionLocalOptimizer instance>') return getattr(self, '__name__', '<FromFunctionLocalOptimizer instance>')
def print_summary(self, stream=sys.stdout, level=0): def print_summary(self, stream=sys.stdout, level=0):
print >> stream, "%s%s id=%i" %(' '*level, print >> stream, "%s%s id=%i" %(' '*level,
str(self.transform), str(self.transform),
...@@ -379,6 +379,9 @@ class LocalOptGroup(LocalOptimizer): ...@@ -379,6 +379,9 @@ class LocalOptGroup(LocalOptimizer):
self.reentrant = any(getattr(opt, 'reentrant', True) for opt in optimizers) self.reentrant = any(getattr(opt, 'reentrant', True) for opt in optimizers)
self.retains_inputs = all(getattr(opt, 'retains_inputs', False) for opt in optimizers) self.retains_inputs = all(getattr(opt, 'retains_inputs', False) for opt in optimizers)
def __str__(self):
return getattr(self, '__name__', '<theano.gof.opt.LocalOptGroup instance>'+str([str(o) for o in self.opts]))
def transform(self, node): def transform(self, node):
for opt in self.opts: for opt in self.opts:
repl = opt.transform(node) repl = opt.transform(node)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论