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)