Function to create custom nn_linear module to overwrite reset_parameters
layer_dense_module(kernel_initializer)
kernel_initializer
string; initializer used to reset_parameters
nn module