Build multiclass and multilabel tf.dataset to train Bert

Hi there,
I’m searching a way to buil a tf.dataset with not just multiple labels but with multiple classes to, but i didn’t find. Is there a way to build this kind of tf.dataset? i’m trying to build a text generator where i can pass multiple arguments to generate a wine description.