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

Python tf.distribute.OneDeviceStrategy.reduce用法及代码示例


用法

reduce(
    reduce_op, value, axis
)

参数

  • reduce_op tf.distribute.ReduceOp 值指定如何组合值。
  • value "per replica" 值，例如由run 返回以组合成一个张量。
  • axis 指定要在每个副本的张量内减少的维度。通常应设置为批量维度，或 None 以仅跨副本减少(例如，如果张量没有批量维度)。

返回

  • 一个Tensor。

跨副本减少 value。

在 OneDeviceStrategy 中，只有一个副本，因此如果axis=None，则简单地返回值。如果轴被指定为 None 以外的值，例如轴 = 0，则值沿该轴减小并返回。

例子：

t = tf.range(10)

result = strategy.reduce(tf.distribute.ReduceOp.SUM, t, axis=None).numpy()
# result:[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

result = strategy.reduce(tf.distribute.ReduceOp.SUM, t, axis=0).numpy()
# result:45

相关用法


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