diff options
author | feiquan <feiquan@wacai.com> | 2018-08-15 08:47:22 +0800 |
---|---|---|
committer | feiquan <feiquan@wacai.com> | 2018-08-15 08:47:22 +0800 |
commit | f2134cbd2ec4dd98f9f20ac41e4f46cdd0246af2 (patch) | |
tree | 83989a84b71d7a08aa9801a573e37c4183f9f64d /tensorflow/contrib/autograph | |
parent | 349d81c80a5b64ae09a36624571ec24d9e7a8b1d (diff) |
use get_item_tensor_string for string with rank 0
Diffstat (limited to 'tensorflow/contrib/autograph')
-rw-r--r-- | tensorflow/contrib/autograph/operators/slices_test.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/tensorflow/contrib/autograph/operators/slices_test.py b/tensorflow/contrib/autograph/operators/slices_test.py index 9c0b2c77a1..5300428462 100644 --- a/tensorflow/contrib/autograph/operators/slices_test.py +++ b/tensorflow/contrib/autograph/operators/slices_test.py @@ -53,6 +53,12 @@ class SlicesTest(test.TestCase): with self.test_session() as sess: self.assertEqual(sess.run(t), b"b") + initial_list_str = constant_op.constant(["abcd", "bcde"]) + t = slices.get_item(initial_list_str, 1, slices.GetItemOpts(element_dtype=initial_str.dtype)) + + with self.test_session() as sess: + self.assertEqual(sess.run(t), b"bcde") + if __name__ == '__main__': test.main() |