class_docstr: Transformer MLP block.
class_docstr_tok: ["Transformer","MLP","block","."]
code: def __init__(self, in_dim, mlp_dim, dropout_rate):
      super().__init__()
      self.linear_1 = nn.Linear(in_dim, mlp_dim)
      self.act = nn.GELU()
      self.dropout_1 = nn.Dropout(dropout_rate)
      self.linear_2 = nn.Linear(mlp_dim, in_dim)
      self.dropout_2 = nn.Dropout(dropout_rate)
      self.init_weights()
  
imports: import copy
  import logging
  import math
  from collections import OrderedDict
  from functools import partial
  from typing import Dict, List, Mapping, NamedTuple, Union
  import torch
  import torch.nn
  from classy_vision.models import ClassyModel, register_model
  from .lecun_normal_init import lecun_normal_init
label_desc: Use image node_id 1 to create a new MLPBlock object from inherited base classes: nn with example: obj = MLPBlock(in_dim, mlp_dim, dropout_rate)