提交 9e403ed0 authored 作者: Yoshua Bengio's avatar Yoshua Bengio

fix to obtain array instead of scalar

上级 452d467c
...@@ -784,7 +784,7 @@ class Subtensor(Op): ...@@ -784,7 +784,7 @@ class Subtensor(Op):
cdata = tuple(map(convert, self.idx_list)) cdata = tuple(map(convert, self.idx_list))
if len(cdata) == 1: if len(cdata) == 1:
cdata = cdata[0] cdata = cdata[0]
out[0] = x.__getitem__(cdata) out[0] = numpy.asarray(x.__getitem__(cdata))
def grad(self, inputs, (gz,)): def grad(self, inputs, (gz,)):
x = inputs[0] x = inputs[0]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论