提交 fc3d3432 authored 作者: Olivier Delalleau's avatar Olivier Delalleau

Fixed bug that suprisingly was not caught earlier

上级 70cbaf4b
......@@ -849,7 +849,7 @@ class TensorFromScalar(Op):
def perform(self, node, (s, ), (out, )):
out[0] = numpy.asarray(s)
def grad(self, (s,), (dt,)):
return [ScalarFromTensor(dt)]
return [scalar_from_tensor(dt)]
tensor_from_scalar = TensorFromScalar()
class ScalarFromTensor(Op):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论