ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

ResourceAccumulatorApplyGradient

public final class ResourceAccumulatorApplyGradient

Applies a gradient to a given accumulator.

Does not add if local_step is lesser than the accumulator's global_step.

Public Methods

static <T> ResourceAccumulatorApplyGradient
create(Scope scope, Operand<?> handle, Operand<Long> localStep, Operand<T> gradient)
Factory method to create a class wrapping a new ResourceAccumulatorApplyGradient operation.

Inherited Methods

Public Methods

public static ResourceAccumulatorApplyGradient create (Scope scope, Operand<?> handle, Operand<Long> localStep, Operand<T> gradient)

Factory method to create a class wrapping a new ResourceAccumulatorApplyGradient operation.

Parameters
scope current scope
handle The handle to a accumulator.
localStep The local_step value at which the gradient was computed.
gradient A tensor of the gradient to be accumulated.
Returns
  • a new instance of ResourceAccumulatorApplyGradient