Conditionally creates batches of tensors based on keep_input
. (deprecated)
tf.compat.v1.train.maybe_batch( tensors, keep_input, batch_size, num_threads=1, capacity=32, enqueue_many=False, shapes=None, dynamic_pad=False, allow_smaller_final_batch=False, shared_name=None, name=None )
See docstring in batch
for more details.
Args | |
---|---|
tensors | The list or dictionary of tensors to enqueue. |
keep_input | A bool Tensor. This tensor controls whether the input is added to the queue or not. If it is a scalar and evaluates True , then tensors are all added to the queue. If it is a vector and enqueue_many is True , then each example is added to the queue only if the corresponding value in keep_input is True . This tensor essentially acts as a filtering mechanism. |
batch_size | The new batch size pulled from the queue. |
num_threads | The number of threads enqueuing tensors . The batching will be nondeterministic if num_threads > 1 . |
capacity | An integer. The maximum number of elements in the queue. |
enqueue_many | Whether each tensor in tensors is a single example. |
shapes | (Optional) The shapes for each example. Defaults to the inferred shapes for tensors . |
dynamic_pad | Boolean. Allow variable dimensions in input shapes. The given dimensions are padded upon dequeue so that tensors within a batch have the same shapes. |
allow_smaller_final_batch | (Optional) Boolean. If True , allow the final batch to be smaller if there are insufficient items left in the queue. |
shared_name | (Optional). If set, this queue will be shared under the given name across multiple sessions. |
name | (Optional) A name for the operations. |
Returns | |
---|---|
A list or dictionary of tensors with the same types as tensors . |
Raises | |
---|---|
ValueError | If the shapes are not specified, and cannot be inferred from the elements of tensors . |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/compat/v1/train/maybe_batch