Mask question

[Alan-Py]

I am dealing with a multi-classification task, where mask is a two-dimensional array, and the numbers in mask represent categories, please ask me what form of mask should I pass in？ can it be 1 0 0 2 0 0 0 1 2 ？ 0is background 1 is class1 2 is class2