Creates a dataset that changes the batch size.
tf.raw_ops.RebatchDatasetV2(
input_dataset, batch_sizes, drop_remainder, output_types, output_shapes,
name=None
)
Creates a dataset that rebatches elements from input_dataset into new batch
sizes.
Args | |
|---|---|
input_dataset
|
A Tensor of type variant.
A variant tensor representing the input dataset.
|
batch_sizes
|
A Tensor of type int64.
A vector of integers representing the size of batches to produce. These values
are cycled through in order.
|
drop_remainder
|
A Tensor of type bool.
|
output_types
|
A list of tf.DTypes that has length >= 1.
|
output_shapes
|
A list of shapes (each a tf.TensorShape or list of ints) that has length >= 1.
|
name
|
A name for the operation (optional). |
Returns | |
|---|---|
A Tensor of type variant.
|