tf.raw_ops.PaddedBatchDataset
Creates a dataset that batches and pads batch_size
elements from the input.
tf.raw_ops.PaddedBatchDataset(
input_dataset,
batch_size,
padded_shapes,
padding_values,
output_shapes,
metadata='',
name=None
)
Args |
input_dataset
|
A Tensor of type variant .
|
batch_size
|
A Tensor of type int64 .
A scalar representing the number of elements to accumulate in a
batch.
|
padded_shapes
|
A list of at least 1 Tensor objects with type int64 .
A list of int64 tensors representing the desired padded shapes
of the corresponding output components. These shapes may be partially
specified, using -1 to indicate that a particular dimension should be
padded to the maximum size of all batch elements.
|
padding_values
|
A list of Tensor objects.
A list of scalars containing the padding value to use for
each of the outputs.
|
output_shapes
|
A list of shapes (each a tf.TensorShape or list of ints ) that has length >= 1 .
|
metadata
|
An optional string . Defaults to "" .
|
name
|
A name for the operation (optional).
|
Returns |
A Tensor of type variant .
|
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-01-23 UTC.
[null,null,["Last updated 2024-01-23 UTC."],[],[]]