當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文

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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。