TensorFlow

API

 tf.distribute / NcclAllReduce


Indicates how a set of values should be reduced.

  • SUM: Add all the values.
  • MEAN: Take the arithmetic mean ("average") of the values.

MEAN tf.distribute.ReduceOp
SUM tf.distribute.ReduceOp

此页内容是否对您有帮助