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. |