提交 a5d33f11 编写于 作者: L LIUJIAN435

add output_type to tf.data.RangeDataset, fix #33414.

上级 60bf8d08
......@@ -884,8 +884,6 @@ class DatasetV2(tracking_base.Trackable, composite_tensor.CompositeTensor):
[]
>>> list(Dataset.range(5, 1, -2).as_numpy_iterator())
[5, 3]
>>> list(Dataset.range(5, output_type=tf.int32).as_numpy_iterator())
[0, 1, 2, 3, 4]
>>> list(Dataset.range(2, 5, output_type=tf.int32).as_numpy_iterator())
[2, 3, 4]
>>> list(Dataset.range(1, 5, 2, output_type=tf.float32).as_numpy_iterator())
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册