提交 8464c20d authored 作者: James Bergstra's avatar James Bergstra

added dtype casting to scalar.mul.grad

上级 b846c08c
...@@ -667,10 +667,10 @@ class Mul(ScalarOp): ...@@ -667,10 +667,10 @@ class Mul(ScalarOp):
retval = [] retval = []
for input in inputs: for input in inputs:
if input.type in grad_types: if input.type in grad_types:
retval += [mul(*([gz] + utils.difference(inputs, [input])))] retval += [cast(mul(*([gz] + utils.difference(inputs, [input]))), input.type.dtype)]
else: else:
retval += [None] retval += [None]
return retval return retval
#return [(mul(*([gz] + utils.difference(inputs, [input]))) #return [(mul(*([gz] + utils.difference(inputs, [input])))
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论