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

Unbatch

public final class Unbatch

Reverses the operation of Batch for a single output Tensor.

An instance of Unbatch either receives an empty batched_tensor, in which case it asynchronously waits until the values become available from a concurrently running instance of Unbatch with the same container and shared_name, or receives a non-empty batched_tensor in which case it finalizes all other concurrently running instances and outputs its own element from the batch.

batched_tensor: The possibly transformed output of Batch. The size of the first dimension should remain unchanged by the transformations for the operation to work. batch_index: The matching batch_index obtained from Batch. id: The id scalar emitted by Batch. unbatched_tensor: The Tensor corresponding to this execution. timeout_micros: Maximum amount of time (in microseconds) to wait to receive the batched input tensor associated with a given invocation of the op. container: Container to control resource sharing. shared_name: Instances of Unbatch with the same container and shared_name are assumed to possibly belong to the same batch. If left empty, the op name will be used as the shared name.

Nested Classes

class Unbatch.Options Optional attributes for Unbatch  

Public Methods

Output<T>
asOutput()
Returns the symbolic handle of a tensor.
static Unbatch.Options
container(String container)
static <T> Unbatch<T>
create(Scope scope, Operand<T> batchedTensor, Operand<Long> batchIndex, Operand<Long> id, Long timeoutMicros, Options... options)
Factory method to create a class wrapping a new Unbatch operation.
static Unbatch.Options
sharedName(String sharedName)
Output<T>

Inherited Methods

Public Methods

public Output<T> 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 Unbatch.Options container (String container)

public static Unbatch<T> create (Scope scope, Operand<T> batchedTensor, Operand<Long> batchIndex, Operand<Long> id, Long timeoutMicros, Options... options)

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

Parameters
scope current scope
options carries optional attributes values
Returns
  • a new instance of Unbatch

public static Unbatch.Options sharedName (String sharedName)

public Output<T> unbatchedTensor ()