Initialize the given weight and bias.
Initialize the given weight and bias.
the weight to initialize
the data format of weight indicating the dimension order of the weight. "output_first" means output is in the lower dimension "input_first" means input is in the lower dimension.
Initialization method to initialize bias and weight. The init method will be called in Module.reset()