[data-stats] Adds support for gathering statistics as metrics with `stats_aggreagtor`.

Also collects metrics for `examples_count`, `features_count`, `feature_values_count`, `feature_lists_count` and `sequence_examples_count`  when `feature_stats()` transformation is applied to the dataset.

PiperOrigin-RevId: 202667632