提交 06d13855 authored 作者: James Bergstra's avatar James Bergstra

tweak to IncSubtensor.str to avoid mislabeling GpuIncSubtensor

上级 19e994e2
......@@ -3539,8 +3539,10 @@ class IncSubtensor(Op):
msg += 'Inc'
else:
msg += 'Set'
return "%s%s{%s}" % (msg,
self.__class__.__name__[3:], ", ".join(indices))
return "%s{%s;%s}" % (
self.__class__.__name__,
msg,
", ".join(indices))
def make_node(self, x, y, *inputs):
x, y = map(as_tensor_variable, [x, y])
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论