public final class RaggedTensorToVariant extends PrimitiveOp implements Operand<Object>
Encodes the given `RaggedTensor` and returns a `variant` Tensor. If `batched_input` is True, then input `RaggedTensor` is unbatched along the zero-th dimension, each component `RaggedTensor` is encoded into a scalar `variant` Tensor, and these are stacked to return a 1-D `variant` Tensor. If `batched_input` is False, then the input `RaggedTensor` is encoded as is and a scalar `variant` Tensor is returned. A `RaggedTensor` is encoded by first creating a 1-D `variant` Tensor with `ragged_rank + 1` elements, containing the splits and values Tensors of the `RaggedTensor`. Then the 1-D `variant` Tensor is wrapped in a scalar `variant` Tensor. See `RaggedTensorFromVariant` for the corresponding decoding logic.
operation
Modifier and Type | Method and Description |
---|---|
Output<Object> |
asOutput()
Returns the symbolic handle of a tensor.
|
static <T extends Number,U> |
create(Scope scope,
Iterable<Operand<T>> rtNestedSplits,
Operand<U> rtDenseValues,
Boolean batchedInput)
Factory method to create a class wrapping a new RaggedTensorToVariant operation.
|
Output<?> |
encodedRagged()
A `variant` Tensor that containing encoded `RaggedTensor`.
|
equals, hashCode, op, toString
public static <T extends Number,U> RaggedTensorToVariant create(Scope scope, Iterable<Operand<T>> rtNestedSplits, Operand<U> rtDenseValues, Boolean batchedInput)
scope
- current scopertNestedSplits
- A list of one or more Tensors representing the splits of the input
`RaggedTensor`.rtDenseValues
- A Tensor representing the values of the input `RaggedTensor`.batchedInput
- A `bool` denoting whether the input is a batched `RaggedTensor`.public Output<?> encodedRagged()
public Output<Object> asOutput()
Operand
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.
asOutput
in interface Operand<Object>
OperationBuilder.addInput(Output)
Copyright © 2022. All rights reserved.