提交 414b1268 authored 作者: Frederic's avatar Frederic

Add Op.__str__ fcts.

上级 cf85d105
......@@ -21,6 +21,11 @@ class DiagonalSubtensor(Op):
"""
Work on the GPU.
"""
def __str__(self):
if self.inplace:
return "%s{inplace}" % self.__class__.__name__
return "%s" % self.__class__.__name__
def __init__(self, inplace):
self.inplace = inplace
if inplace:
......@@ -57,6 +62,11 @@ diagonal_subtensor = DiagonalSubtensor(False)
class IncDiagonalSubtensor(Op):
def __str__(self):
if self.inplace:
return "%s{inplace}" % self.__class__.__name__
return "%s" % self.__class__.__name__
def __init__(self, inplace):
self.inplace = inplace
if inplace:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论