class_docstr: Multi-layer perception module.
class_docstr_tok: ["Multi-layer","perception","module","."]
code: def forward(self, x):
      x = self.fc1(x)
      x = self.act(x)
      if self.drop_rate > 0.0:
          x = self.drop(x)
      x = self.fc2(x)
      if self.drop_rate > 0.0:
          x = self.drop(x)
      return x
  
imports: import math
  from functools import partial
  from typing import Tuple, List
  import numpy
  import torch
  from torch import nn
  from torch.nn.init import trunc_normal_
  from einops import rearrange
  from towhee.models.layers.layers_with_relprop import GELU, Linear, Dropout, Conv2d, Conv3d, Softmax, Sigmoid, LayerNorm, Einsum, Add, Clone, MaxPool3d, IndexSelect
label_desc: Use image node_id 2 for calling the Mlp obj's underlying member method code with example usage: obj.forward(x) and returns: x