Creates a dataset that changes the batch size.
tf.raw_ops.RebatchDataset(
input_dataset,
num_replicas,
output_types,
output_shapes,
use_fallback=True,
name=None
)
Creates a dataset that changes the batch size of the dataset to current batch size // num_workers.
Args | |
---|---|
input_dataset
|
A Tensor of type variant .
A variant tensor representing the input dataset.
|
num_replicas
|
A Tensor of type int64 .
A scalar representing the number of replicas to distribute this batch across. As
a result of this transformation the current batch size would end up being
divided by this parameter.
|
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 .
|
use_fallback
|
An optional bool . Defaults to True .
|
name
|
A name for the operation (optional). |
Returns | |
---|---|
A Tensor of type variant .
|