class_docstr: This class implements the `MobileViT architecture <https://arxiv.org/abs/2110.02178?context=cs.LG>`_
class_docstr_tok: ["This","class","implements","the","`","MobileViT","architecture","<","https",":","\/\/arxiv.org\/abs\/2110.02178","?","context=cs.LG",">","`","_"]
code: def forward(self, x: Tensor) -> Tensor:
      x = self.conv_1(x)
      x = self.layer_1(x)
      x = self.layer_2(x)
  
      x = self.layer_3(x)
      x = self.layer_4(x)
      x = self.layer_5(x)
      x = self.conv_1x1_exp(x)
      x = self.classifier(x)
      return x
  
imports: from typing import Optional, Tuple, Union, Dict
  import math
  import torch
  import torch.nn
  from torch import Tensor
  from torch.nn import functional
  from transformer import TransformerEncoder
  from model_config import get_config
label_desc: Use image node_id 6 for calling the MobileViT obj's underlying member method code with example usage: obj.forward(x) and returns: x