tf.data.Dataset. The tf.data.Dataset object to visualize. Examples
should not be batched. Examples will be consumed in order until (rows *
cols) are read or the dataset is consumed.
ds_info
The dataset info object to which extract the label and features
info. Available either through tfds.load('mnist', with_info=True) or
tfds.builder('mnist').info