提交 96f4e9f3 authored 作者: John Schulman's avatar John Schulman

mod derivative

上级 f3f3b80a
......@@ -1655,8 +1655,8 @@ class Mod(BinaryScalarOp):
""") % locals()
def grad(self, (x, y), (gz, )):
return [x.zeros_like(dtype=theano.config.floatX),
y.zeros_like(dtype=theano.config.floatX)]
return [gz,
-(x//y) * gz ]
mod = Mod(upcast_out, name='mod')
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论