code: def get_prompt_token_labels_for_megatron_gpt(
      self, input_ids, num_prompt_tokens
  ):
      prompt_token_labels = torch.full(
          size=(input_ids.size(0), num_prompt_tokens),
          fill_value=self.tokenizer.tokenizer.pad_token_id,
          dtype=torch.long,
      )
  
      if self.prompt_learning:
          prompt_token_labels.data = torch.LongTensor(
              np.tile(
                  np.array(self.language_model.pseudo_token_ids),
                  (input_ids.size(0), 1),
              )
          )
  
      prompt_token_labels = prompt_token_labels.to(input_ids.device)
  
      return prompt_token_labels
  
imports: import collections
  import copy
  import os
  import random
  from typing import Dict, Optional, Union
  import numpy
  import torch
  from omegaconf import DictConfig
  from pytorch_lightning import Trainer
  from torch.utils.data import DataLoader
  from transformers import AutoModelWithLMHead
  from nemo.collections.nlp.data.dialogue import DialogueGPTClassificationDataset, DialogueSGDDataProcessor
  from nemo.collections.nlp.data.dialogue.data_processor.assistant_data_processor import DialogueAssistantDataProcessor
  from nemo.collections.nlp.data.dialogue.data_processor.design_data_processor import DialogueDesignDataProcessor
  from nemo.collections.nlp.metrics.classification_report import ClassificationReport
  from nemo.collections.nlp.metrics.dialogue_metrics import DialogueClassificationMetrics
  from nemo.collections.nlp.models.language_modeling.megatron_gpt_model import MegatronGPTModel
  from nemo.collections.nlp.models.language_modeling.megatron_gpt_prompt_learning_model import MegatronGPTPromptLearningModel
  from nemo.collections.nlp.models.nlp_model import NLPModel
  from nemo.collections.nlp.modules.common import VirtualPromptSource, VirtualPromptStyle
  from nemo.collections.nlp.modules.common.text_generation_utils import get_default_sampling_params, megatron_gpt_generate
  from nemo.collections.nlp.modules.common.transformer.text_generation import LengthParam
  from nemo.collections.nlp.parts.nlp_overrides import NLPSaveRestoreConnector
  from nemo.core.classes.common import PretrainedModelInfo
  from nemo.utils import logging
label_desc: Use image node_id 11 for calling the DialogueGPTClassificationModel obj's underlying member method code with example usage: obj.get_prompt_token_labels_for_megatron_gpt(input_ids, num_prompt_tokens) and returns: prompt_token_labels