提交 1a69c034 authored 作者: Olivier Breuleux's avatar Olivier Breuleux

fixed scalar.Identity gradient bug

上级 bf17e33b
...@@ -291,7 +291,7 @@ class Identity(UnaryScalarOp): ...@@ -291,7 +291,7 @@ class Identity(UnaryScalarOp):
return x return x
def c_code(self, (x, ), (z, ), sub): def c_code(self, (x, ), (z, ), sub):
return "%(z)s = %(x)s;" % locals() return "%(z)s = %(x)s;" % locals()
def grad(self, (x, y), (gz, )): def grad(self, (x, ), (gz, )):
return gz, return gz,
class Neg(UnaryScalarOp): class Neg(UnaryScalarOp):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论