class_docstr: Just  an MLP
class_docstr_tok: ["Just","an","MLP"]
code: def __init__(
      self, n_inputs, n_outputs, mlp_width, mlp_depth, mlp_dropout
  ):
      super(MLP, self).__init__()
      self.input = nn.Linear(n_inputs, mlp_width)
      self.dropout = nn.Dropout(mlp_dropout)
      self.hiddens = nn.ModuleList(
          [
              nn.Linear(mlp_width, mlp_width)
              for _ in range(mlp_depth - 2)
          ]
      )
      self.output = nn.Linear(mlp_width, n_outputs)
      self.n_outputs = n_outputs
  
imports: import torch
  import torch.nn
  import torch.nn.functional
  import torchvision.models
label_desc: Use image node_id 1 to create a new MLP object from inherited base classes: nn with example: obj = MLP(n_inputs, n_outputs, mlp_width, mlp_depth, mlp_dropout)