tf.data.experimental.group_by_window