org.tensorflow.op.sparse

Class SparseAccumulatorTakeGradient<T>

java.lang.Object

org.tensorflow.op.PrimitiveOp

org.tensorflow.op.sparse.SparseAccumulatorTakeGradient<T>

Type Parameters:

T - data type for values() output

All Implemented Interfaces:

Op

@Operator(group="sparse")
public final class SparseAccumulatorTakeGradient<T>
extends PrimitiveOp

Extracts the average sparse gradient in a SparseConditionalAccumulator.

The op will blocks until sufficient (i.e., more than num_required) gradients have been accumulated. If the accumulator has already aggregated more than num_required gradients, it will return its average of the accumulated gradients. Also automatically increments the recorded global_step in the accumulator by 1, and resets the aggregate to 0.

Field Summary

Fields inherited from class org.tensorflow.op.PrimitiveOp

operation

Method Summary

Modifier and Type	Method and Description
static <T> SparseAccumulatorTakeGradient<T>	create(Scope scope, Operand<String> handle, Operand<Integer> numRequired, Class<T> dtype) Factory method to create a class wrapping a new SparseAccumulatorTakeGradient operation.
Output<Long>	indices() Indices of the average of the accumulated sparse gradients.
Output<Long>	shape() Shape of the average of the accumulated sparse gradients.
Output<T>	values() Values of the average of the accumulated sparse gradients.

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> SparseAccumulatorTakeGradient<T> create(Scope scope,
                                                          Operand<String> handle,
                                                          Operand<Integer> numRequired,
                                                          Class<T> dtype)

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

Parameters:

scope - current scope

handle - The handle to a SparseConditionalAccumulator.

numRequired - Number of gradients required before we return an aggregate.

dtype - The data type of accumulated gradients. Needs to correspond to the type of the accumulator.

Returns:

a new instance of SparseAccumulatorTakeGradient

indices

public Output<Long> indices()

Indices of the average of the accumulated sparse gradients.

values

public Output<T> values()

Values of the average of the accumulated sparse gradients.

shape

public Output<Long> shape()

Shape of the average of the accumulated sparse gradients.