提交 5594fd25 authored 作者: notoraptor's avatar notoraptor

Fix test:

theano/gpuarray/tests/test_subtensor.py:G_subtensor.test_noncontiguous_idx
上级 b7890b77
......@@ -533,7 +533,7 @@ class T_subtensor(unittest.TestCase, utt.TestOptimizationMixin):
data = rand(4, 2, 3)
idx = [2, 2, 0, 0, 1, 1]
n = self.shared(data)
t = n[self.shared(numpy.asarray(idx))[::2]]
t = n[self.shared(numpy.asarray(idx).astype('int64'))[::2]]
self.assertTrue(isinstance(t.owner.op, tensor.AdvancedSubtensor1))
val = self.eval_output_and_check(t, op_type=self.adv_sub1, length=2)
utt.assert_allclose(data[idx[::2]], val)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论