提交 98ccc988 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Make sure to always return arrays.

上级 6a2fa0a8
......@@ -1218,7 +1218,7 @@ class UnravelIndex(gof.Op):
res = np.unravel_index(indices, dims)
assert len(res) == len(out)
for i in xrange(len(out)):
out[i][0] = res[i]
out[i][0] = theano._asarray(res[i])
def unravel_index(indices, dims, order='C', ndim=None):
......@@ -1304,8 +1304,9 @@ class RavelMultiIndex(gof.Op):
def perform(self, node, inp, out):
multi_index, dims = inp[:-1], inp[-1]
out[0][0] = np.ravel_multi_index(multi_index, dims,
res = np.ravel_multi_index(multi_index, dims,
mode=self.mode, order=self.order)
out[0][0] = theano._asarray(res)
def ravel_multi_index(multi_index, dims, mode='raise', order='C'):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论