class_docstr: Launch MLP in C++
  
  Args:
      mlp_sizes (list of int): MLP sizes. Example: [1024,1024,1024] will create 2 MLP layers with shape 1024x1024
      bias (bool): Default True:
      relu (bool): Default True
class_docstr_tok: ["Launch","MLP","in","C++","Args",":","mlp_sizes","(","list","of","int",")",":","MLP","sizes",".","Example",":","[","1024,1024,1024","]","will","create","2","MLP","layers","with","shape","1024x1024","bias","(","bool",")",":","Default","True",":","relu","(","bool",")",":","Default","True"]
code: def reset_parameters(self):
      for weight in self.weights:
          dimsum = weight.size(0) + weight.size(1)
          std = math.sqrt(2.0 / float(dimsum))
          nn.init.normal_(weight, 0.0, std)
      if self.bias:
          for bias in self.biases:
              std = math.sqrt(1.0 / float(bias.size(0)))
              nn.init.normal_(bias, 0.0, std)
  
file_path: apex/apex/mlp/mlp.py
imports: from copy import copy
  import math
  import torch
  from torch import nn
  from apex._autocast_utils import _cast_if_autocast_enabled
  import mlp_cuda
label_desc: Use image node_id 2 for calling the MLP obj's underlying member method code with example usage: obj.reset_parameters() without return types