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

Fixed bug that suprisingly was not caught earlier

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