chunked/nltk_chunking/_glossary/chunk_80.txt

For semantic segmentation models, ([SegformerForSemanticSegmentation]), the model expects a tensor of dimension
  (batch_size, height, width) with each value of the batch corresponding to the expected label of each individual pixel.