What you are doing here seems correct
just make sure when you are passing labels it’s value should be between [0.. num_choices-1]
labels
[0.. num_choices-1]