Creates a dataset that changes the batch size.

    input_dataset, num_replicas, output_types, output_shapes, use_fallback=True,

Creates a dataset that changes the batch size of the dataset to current batch size // num_replicas.


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


A Tensor of type variant.