Python tf.data.Iterator.get_next_as_optional用法及代码示例

用法

@abc.abstractmethod
get_next_as_optional()

返回包装在 tf.experimental.Optional 中的下一个元素。

如果迭代器已到达序列末尾，则返回的tf.experimental.Optional 将没有值。

dataset = tf.data.Dataset.from_tensors(42)
iterator = iter(dataset)
optional = iterator.get_next_as_optional()
print(optional.has_value())
tf.Tensor(True, shape=(), dtype=bool)
print(optional.get_value())
tf.Tensor(42, shape=(), dtype=int32)
optional = iterator.get_next_as_optional()
print(optional.has_value())
tf.Tensor(False, shape=(), dtype=bool)