tf.data.dataset how to get total elements code example

Example: how to know the length of a dataset tensorflow

ds = tf.data.TFRecordDataset(dataset_filenames)
ds_size = sum(1 for _ in ds)