def testEnumerateDataset(self):
components = (["a", "b"], [1, 2], [37.0, 38])
start = constant_op.constant(20, dtype=dtypes.int64)
iterator = (dataset_ops.Dataset.from_tensor_slices(components).apply(
enumerate_ops.enumerate_dataset(start)).make_initializable_iterator())
init_op = iterator.initializer
get_next = iterator.get_next()
self.assertEqual(dtypes.int64, get_next[0].dtype)
self.assertEqual((), get_next[0].shape)
self.assertEqual([tensor_shape.TensorShape([])] * 3,
[t.shape for t in get_next[1]])
with self.test_session() as sess:
sess.run(init_op)
self.assertEqual((20, (b"a", 1, 37.0)), sess.run(get_next))
self.assertEqual((21, (b"b", 2, 38.0)), sess.run(get_next))
with self.assertRaises(errors.OutOfRangeError):
sess.run(get_next)
def enumerate(self, start=0):
"""Deprecated: Use `Dataset.apply(tf.contrib.data.enumerate_dataset(..)`."""
return self.apply(enumerate_ops.enumerate_dataset(start))
def enumerate(self, start=0):
"""Deprecated: Use `Dataset.apply(tf.contrib.data.enumerate_dataset(..)`."""
return self.apply(enumerate_ops.enumerate_dataset(start))
