提交 92327dec authored 作者: Frederic's avatar Frederic

Allow to create the op with default parameter value.

上级 754dde21
......@@ -26,7 +26,7 @@ class DiagonalSubtensor(Op):
return "%s{inplace}" % self.__class__.__name__
return "%s" % self.__class__.__name__
def __init__(self, inplace):
def __init__(self, inplace=False):
self.inplace = inplace
if inplace:
self.view_map = {0: [0]}
......@@ -67,7 +67,7 @@ class IncDiagonalSubtensor(Op):
return "%s{inplace}" % self.__class__.__name__
return "%s" % self.__class__.__name__
def __init__(self, inplace):
def __init__(self, inplace=False):
self.inplace = inplace
if inplace:
self.destroy_map = {0: [0]}
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论