class_docstr: Vision Transformer
  A PyTorch impl of : `An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale`
      - https://arxiv.org/abs/2010.11929
class_docstr_tok: ["Vision","Transformer","A","PyTorch","impl","of",":","`","An","Image","is","Worth","16x16","Words",":","Transformers","for","Image","Recognition","at","Scale","`","-","https",":","\/\/arxiv.org\/abs\/2010.11929"]
code: def forward_head(self, x, pre_logits: bool = False):
      if self.global_pool:
          x = x[:, 0]
      return x if pre_logits else self.head(x)
  
imports: import math
  from functools import partial
  from copy import deepcopy
  import torch
  import torch.nn
  import torch.nn.functional
  from torch.nn.init import trunc_normal_
  from timm.models.helpers import named_apply
  from timm.models.layers import PatchEmbed
  from src.models.modules.mha import MHA
  from src.models.modules.mlp import Mlp, FusedDenseGeluDense
  from src.models.modules.block import Block
label_desc: Use image node_id 7 for calling the VisionTransformer obj's underlying member method code with example usage: obj.forward_head(x, pre_logits) and returns: unknown