提交 853a5915 authored 作者: Olivier Breuleux's avatar Olivier Breuleux

fixed gradient in tensor_random

上级 92e42e6c
...@@ -133,7 +133,7 @@ class NumpyGenerator(gof.op.Op): ...@@ -133,7 +133,7 @@ class NumpyGenerator(gof.op.Op):
return gof.Apply(op = self, inputs = inputs, outputs = outputs) return gof.Apply(op = self, inputs = inputs, outputs = outputs)
def grad(self, inputs, grad_outputs): def grad(self, inputs, grad_outputs):
return [None] return [None, None]
def perform(self, node, input_storage, output_storage): def perform(self, node, input_storage, output_storage):
rng = input_storage[0] rng = input_storage[0]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论