当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


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。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。