Python tf.data.Dataset.reduce用法及代码示例

用法

reduce(
    initial_state, reduce_func, name=None
)

initial_state 表示转换初始状态的元素。
reduce_func 将 (old_state, input_element) 映射到 new_state 的函数。它必须接受两个参数并返回一个新元素 new_state 的结构必须与 initial_state 的结构匹配。
name (可选。) tf.data 操作的名称。

将输入数据集缩减为单个元素。

转换在输入数据集的每个元素上连续调用reduce_func，直到数据集用尽，在其内部状态中聚合信息。 initial_state 参数用于初始状态，最终状态作为结果返回。

tf.data.Dataset.range(5).reduce(np.int64(0), lambda x, _:x + 1).numpy()
5
tf.data.Dataset.range(5).reduce(np.int64(0), lambda x, y:x + y).numpy()
10

相关用法

注：本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.data.Dataset.reduce。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。