tf.data.Dataset: how to get the dataset size (number of elements in a epoch)? tf.data.Dataset: how to get the dataset size (number of elements in a epoch)? python-3.x python-3.x