提交 4c42fbff authored 作者: Frederic's avatar Frederic

make curand op have a more readable string representation.

上级 1cbb3cb8
......@@ -64,6 +64,9 @@ class CURAND_Base(theano.gof.Op):
def __hash__(self):
return hash((type(self), self._config()))
def __str__(self):
return self.__class__.__name__+"{inplace=%s, out_dtype=%s}"%(
self.destructive, self.output_type)
def make_node(self, generator, size):
return theano.gof.Apply(self, [generator, size],
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论