Get length of a dataset in Tensorflow Get length of a dataset in Tensorflow python-3.x python-3.x