src/transformers/models/mask2former/modeling_mask2former.py/Mask2FormerForUniversalSegmentation/get_auxiliary_logits
class Mask2FormerForUniversalSegmentation:    def get_auxiliary_logits(self, classes: torch.Tensor, output_masks: torch.Tensor):
        auxiliary_logits: List[Dict(str, Tensor)] = []

        for aux_binary_masks, aux_classes in zip(output_masks[:-1], classes[:-1]):
            auxiliary_logits.append({"masks_queries_logits": aux_binary_masks, "class_queries_logits": aux_classes})

        return auxiliary_logits