code: def mlp(hiddens=[], layer_norm=False):
      
      return lambda *args, **kwargs: _mlp(hiddens, layer_norm=layer_norm, *args, **kwargs)
docs: This model takes as input an observation and returns values of all actions.
  
      Parameters
      ----------
      hiddens: [int]
          list of sizes of hidden layers
      layer_norm: bool
          if true applies layer normalization for every layer
          as described in https://arxiv.org/abs/1607.06450
  
      Returns
      -------
      q_func: function
          q_function for DQN algorithm.