code: def __init__(
      self,
      model_size: int,
      vocab_target_size: int,
      model_type: str = C.NVS_TYPE_LOGIT_MAX,
      dtype: Optional[pt.dtype] = None,
  ) -> None:
      super().__init__()
      self.vocab_target_size = vocab_target_size
      self.model_type = model_type
  
      self.project_vocab = pt.nn.Linear(
          model_size, vocab_target_size, bias=True, dtype=dtype
      )
  
imports: from typing import Optional
  import torch
  from .None import constants
label_desc: Use image node_id 1 to create a new NeuralVocabSelection object from inherited base classes: pt.nn with example: obj = NeuralVocabSelection(model_size, vocab_target_size, model_type, dtype)