I set output_dim=10. I gave ‘10’ value because of the following.
Consider the following sentence,
Andrew      ng      is  a  data scientist
B-PER     I-PER  O  O   O     O
For each token, it can have any of the 10 different tags like B-PER, I-PER, B-LOC, I-LOC.