提交 152c46b6 authored 作者: Ying Zhang's avatar Ying Zhang

small changes

上级 ba70cb82
......@@ -1541,7 +1541,7 @@ class Assert(T.Op):
return hash(type(self)) ^ hash(self.msg)
def grad(self, input, output_gradients):
return output_gradients + [DisconnectedType()()] * (len(inputs) - 1)
return output_gradients + [DisconnectedType()()] * (len(input) - 1)
def c_code(self, node, name, inames, onames, sub):
value = inames[0]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论