# tf.raw_ops.ExperimentalRebatchDataset

<br />

Creates a dataset that changes the batch size.

#### View aliases


**Compat aliases for migration**

See
[Migration guide](https://www.tensorflow.org/guide/migrate) for
more details.

[`tf.compat.v1.raw_ops.ExperimentalRebatchDataset`](https://www.tensorflow.org/api_docs/python/tf/raw_ops/ExperimentalRebatchDataset)

<br />

    tf.raw_ops.ExperimentalRebatchDataset(
        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_replicas.

<br />

<br />

<br />

| ## 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`](https://www.tensorflow.org/versions/r2.13/api_docs/python/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). |

<br />

<br />

<br />

<br />

| ## Returns ||
|---|---|
| A `Tensor` of type `variant`. ||

<br />