org.tensorflow.op.core

Class SparseAccumulatorApplyGradient

java.lang.Object

org.tensorflow.op.PrimitiveOp

org.tensorflow.op.core.SparseAccumulatorApplyGradient

All Implemented Interfaces:

Op

@Operator
public final class SparseAccumulatorApplyGradient
extends PrimitiveOp

Applies a sparse gradient to a given accumulator.

Does not add if local_step is smaller 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> SparseAccumulatorApplyGradient	create(Scope scope, Operand<String> handle, Operand<Long> localStep, Operand<Long> gradientIndices, Operand<T> gradientValues, Operand<Long> gradientShape, Boolean hasKnownShape) Factory method to create a class to wrap a new SparseAccumulatorApplyGradient operation to the graph.

Methods inherited from class org.tensorflow.op.PrimitiveOp

equals, hashCode, toString

Methods inherited from class java.lang.Object

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

Method Detail

create

public static <T> SparseAccumulatorApplyGradient create(Scope scope,
                                                        Operand<String> handle,
                                                        Operand<Long> localStep,
                                                        Operand<Long> gradientIndices,
                                                        Operand<T> gradientValues,
                                                        Operand<Long> gradientShape,
                                                        Boolean hasKnownShape)

Factory method to create a class to wrap a new SparseAccumulatorApplyGradient operation to the graph.

Parameters:

scope - current graph scope

handle - The handle to a accumulator.

localStep - The local_step value at which the sparse gradient was computed.

gradientIndices - Indices of the sparse gradient to be accumulated. Must be a vector.

gradientValues - Values are the non-zero slices of the gradient, and must have the same first dimension as indices, i.e., the nnz represented by indices and values must be consistent.

gradientShape - Shape of the sparse gradient to be accumulated.

hasKnownShape - Boolean indicating whether gradient_shape is unknown, in which case the input is ignored during validation.

Returns:

a new instance of SparseAccumulatorApplyGradient