T
- data type for unbatchedTensor()
output@Operator public final class Unbatch<T> extends PrimitiveOp implements Operand<T>
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.
Modifier and Type | Class and Description |
---|---|
static class |
Unbatch.Options
Optional attributes for
Unbatch |
operation
Modifier and Type | Method and Description |
---|---|
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,
Unbatch.Options... options)
Factory method to create a class wrapping a new Unbatch operation.
|
static Unbatch.Options |
sharedName(String sharedName) |
Output<T> |
unbatchedTensor() |
equals, hashCode, op, toString
public static <T> Unbatch<T> create(Scope scope, Operand<T> batchedTensor, Operand<Long> batchIndex, Operand<Long> id, Long timeoutMicros, Unbatch.Options... options)
scope
- current scopebatchedTensor
- batchIndex
- id
- timeoutMicros
- options
- carries optional attributes valuespublic static Unbatch.Options container(String container)
container
- public static Unbatch.Options sharedName(String sharedName)
sharedName
- public Output<T> 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<T>
OperationBuilder.addInput(Output)
Copyright © 2022. All rights reserved.