class_docstr: Implementation of Metaformer block with RepMixer as token mixer.
  
  For more details on Metaformer structure, please refer to:
  `MetaFormer Is Actually What You Need for Vision <https://arxiv.org/pdf/2111.11418.pdf>`_
class_docstr_tok: ["Implementation","of","Metaformer","block","with","RepMixer","as","token","mixer",".","For","more","details","on","Metaformer","structure",",","please","refer","to",":","`","MetaFormer","Is","Actually","What","You","Need","for","Vision","<","https",":","\/\/arxiv.org\/pdf\/2111.11418.pdf",">","`","_"]
code: def forward(self, x):
      if self.use_layer_scale:
          x = self.token_mixer(x)
          x = x + self.drop_path(self.layer_scale * self.convffn(x))
      else:
          x = self.token_mixer(x)
          x = x + self.drop_path(self.convffn(x))
      return x
  
imports: import os
  import copy
  from functools import partial
  from typing import List, Tuple, Optional, Union
  import torch
  import torch.nn
  from timm.data import IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD
  from timm.models.layers import DropPath, trunc_normal_
  from timm.models.registry import register_model
  from models.modules.mobileone import MobileOneBlock
  from models.modules.replknet import ReparamLargeKernelConv
label_desc: Use image node_id 2 for calling the RepMixerBlock obj's underlying member method code with example usage: obj.forward(x) and returns: x