tensorflow::ops::ConditionalAccumulator

tensorflow::ops::ConditionalAccumulator

#include <data_flow_ops.h>

A conditional accumulator for aggregating gradients.

Summary

The accumulator accepts gradients marked with local_step greater or equal to the most recent global_step known to the accumulator. The average can be extracted from the accumulator, provided sufficient gradients have been accumulated. Extracting the average automatically resets the aggregate to 0, and increments the global_step recorded by the accumulator.

Arguments:

  • scope: A Scope object
  • dtype: The type of the value being accumulated.
  • shape: The shape of the values, can be [], in which case shape is unknown.

Optional attributes (see 登录查看完整内容