org.tensorflow.op.train

Class AccumulatorApplyGradient

java.lang.Object

org.tensorflow.op.PrimitiveOp

org.tensorflow.op.train.AccumulatorApplyGradient

All Implemented Interfaces:

Op

@Operator(group="train")
public final class AccumulatorApplyGradient
extends PrimitiveOp

Applies a gradient to a given accumulator.

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

Field Summary

Fields inherited from class org.tensorflow.op.PrimitiveOp

operation

Method Summary

Modifier and Type	Method and Description
static <T> AccumulatorApplyGradient	create(Scope scope, Operand<String> handle, Operand<Long> localStep, Operand<T> gradient) Factory method to create a class wrapping a new AccumulatorApplyGradient operation.

Methods inherited from class org.tensorflow.op.PrimitiveOp

equals, hashCode, op, toString

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Method Detail

create

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

Factory method to create a class wrapping a new AccumulatorApplyGradient 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 AccumulatorApplyGradient