tf.compat.v1.train.maybe_batch | TensorFlow v2.16.1 (original) (raw)
tf.compat.v1.train.maybe_batch
Stay organized with collections Save and categorize content based on your preferences.
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, thentensors are all added to the queue. If it is a vector and enqueue_manyis 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. |
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.
Last updated 2024-04-26 UTC.