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

Make sure to always return arrays.

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