tf.data.experimental.unbatch

Splits elements of a dataset into multiple elements on the batch dimension.

For example, if elements of the dataset are shaped [B, a0, a1, ...],
where B may vary for each input element, then for each element in the
dataset, the unbatched dataset will contain B consecutive elements
of shape [a0, a1, ...].