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

RaggedTensorToVariantGradient

public final class RaggedTensorToVariantGradient

Helper used to compute the gradient for `RaggedTensorToVariant`.

Computes the gradient for the dense_values input to the RaggedTensorToVariant op, given the variant-encoded ragged gradients of the outputs, along with the outer row-splits and the shape of the dense-values that were provided as inputs to the RaggedTensorToVariant op.

Public Methods

Output<U>
asOutput()
Returns the symbolic handle of a tensor.
static <U, T extends Number> RaggedTensorToVariantGradient<U>
create(Scope scope, Operand<?> encodedRaggedGrad, Operand<T> rowSplits, Operand<Integer> denseValuesShape, Class<U> Tvalues)
Factory method to create a class wrapping a new RaggedTensorToVariantGradient operation.
Output<U>
denseValuesGrad()
Gradient for the dense_values of the RaggedTensorToVariant op.

Inherited Methods

Public Methods

public Output<U> asOutput ()

Returns the symbolic handle of a tensor.

Inputs to TensorFlow operations are outputs of another TensorFlow operation. This method is used to obtain a symbolic handle that represents the computation of the input.

public static RaggedTensorToVariantGradient<U> create (Scope scope, Operand<?> encodedRaggedGrad, Operand<T> rowSplits, Operand<Integer> denseValuesShape, Class<U> Tvalues)

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

Parameters
scope current scope
encodedRaggedGrad A `variant` Tensor containing encoded `RaggedTensor` gradients.
rowSplits Outermost row-splits that were used as input to the RaggedTensorToVariant op.
denseValuesShape Shape of the dense_values that was used as an input to the RaggedTensorToVariant op.
Returns
  • a new instance of RaggedTensorToVariantGradient

public Output<U> denseValuesGrad ()

Gradient for the dense_values of the RaggedTensorToVariant op.