Helper used to compute the gradient for RaggedTensorToVariant.
tf.raw_ops.RaggedTensorToVariantGradient(
encoded_ragged_grad, row_splits, dense_values_shape, Tvalues, name=None
)
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.
Args | |
|---|---|
encoded_ragged_grad
|
A Tensor of type variant.
A variant Tensor containing encoded RaggedTensor gradients.
|
row_splits
|
A Tensor. Must be one of the following types: int32, int64.
Outermost row-splits that were used as input to the RaggedTensorToVariant op.
|
dense_values_shape
|
A Tensor of type int32.
Shape of the dense_values that was used as an input to the
RaggedTensorToVariant op.
|
Tvalues
|
A tf.DType.
|
name
|
A name for the operation (optional). |
Returns | |
|---|---|
A Tensor of type Tvalues.
|