提交 0ef33a81 authored 作者: abalkin's avatar abalkin

Trying to fix travis build failures. Make indices explicit 32-bit int array.

上级 4f2492f2
...@@ -7064,7 +7064,7 @@ class TestTensorInstanceMethods(unittest.TestCase): ...@@ -7064,7 +7064,7 @@ class TestTensorInstanceMethods(unittest.TestCase):
assert_array_equal(X.take(indices).eval({X: x}), x.take(indices)) assert_array_equal(X.take(indices).eval({X: x}), x.take(indices))
indices = [1,0,1] indices = [1,0,1]
assert_array_equal(X.take(indices, 1).eval({X: x}), x.take(indices, 1)) assert_array_equal(X.take(indices, 1).eval({X: x}), x.take(indices, 1))
indices = [-10,5,12] indices = numpy.array([-10,5,12], dtype='int32')
assert_array_equal(X.take(indices, 1, mode='wrap').eval({X: x}), assert_array_equal(X.take(indices, 1, mode='wrap').eval({X: x}),
x.take(indices, 1, mode='wrap')) x.take(indices, 1, mode='wrap'))
assert_array_equal(X.take(indices, -1, mode='wrap').eval({X: x}), assert_array_equal(X.take(indices, -1, mode='wrap').eval({X: x}),
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论