class_docstr: Module MobileViT. Default arguments is for MobileViT XXS.
  
  Paper: https://arxiv.org/abs/2110.02178
  Based on: https://github.com/chinhsuanwu/mobilevit-pytorch
  
  Args:
      mode: 'xxs', 'xs' or 's', defaults to 'xxs'.
      in_channels: the number of channels for the input image.
      patch_size: image_size must be divisible by patch_size.
      dropout: dropout ratio in Transformer.
  
  Example:
      >>> img = torch.rand(1, 3, 256, 256)
      >>> mvit = MobileViT(mode='xxs')
      >>> mvit(img).shape
      torch.Size([1, 320, 8, 8])
class_docstr_tok: ["Module","MobileViT",".","Default","arguments","is","for","MobileViT","XXS",".","Paper",":","https",":","\/\/arxiv.org\/abs\/2110.02178","Based","on",":","https",":","\/\/github.com\/chinhsuanwu\/mobilevit-pytorch","Args",":","mode",":","'xxs","'",",","'xs","'","or","'s","'",",","defaults","to","'xxs","'",".","in_channels",":","the","number","of","channels","for","the","input","image",".","patch_size",":","image_size","must","be","divisible","by","patch_size",".","dropout",":","dropout","ratio","in","Transformer",".","Example",":",">",">",">","img","=","torch.rand","(","1",",","3",",","256",",","256",")",">",">",">","mvit","=","MobileViT","(","mode='xxs","'",")",">",">",">","mvit","(","img",")",".shape","torch.Size","(","[","1",",","320",",","8",",","8","]",")"]
code: def forward(self, x: Tensor) -> Tensor:
      x = self.conv1(x)
      x = self.mv2[0](x)
  
      x = self.mv2[1](x)
      x = self.mv2[2](x)
      x = self.mv2[3](x)  # Repeat
  
      x = self.mv2[4](x)
      x = self.mvit[0](x)
  
      x = self.mv2[5](x)
      x = self.mvit[1](x)
  
      x = self.mv2[6](x)
      x = self.mvit[2](x)
      x = self.conv2(x)
      return x
  
imports: from typing import Any, Dict, Tuple
  import torch
  from torch import nn
  from kornia.core import Module, Tensor
label_desc: Use image node_id 2 for calling the MobileViT obj's underlying member method code with example usage: obj.forward(x) and returns: x