class_docstr: LSTM Model.
  
  Args:
      output_dim (int): Dimension of the network output.
      hidden_dim (int): Hidden dimension for LSTM cell.
      name (str): Policy name, also the variable scope.
      hidden_nonlinearity (callable): Activation function for intermediate
          dense layer(s). It should return a tf.Tensor. Set it to
          None to maintain a linear activation.
      hidden_w_init (callable): Initializer function for the weight
          of intermediate dense layer(s). The function should return a
          tf.Tensor.
      hidden_b_init (callable): Initializer function for the bias
          of intermediate dense layer(s). The function should return a
          tf.Tensor.
      recurrent_nonlinearity (callable): Activation function for recurrent
          layers. It should return a tf.Tensor. Set it to None to
          maintain a linear activation.
      recurrent_w_init (callable): Initializer function for the weight
          of recurrent layer(s). The function should return a
          tf.Tensor.
      output_nonlinearity (callable): Activation function for output dense
          layer. It should return a tf.Tensor. Set it to None to
          maintain a linear activation.
      output_w_init (callable): Initializer function for the weight
          of output dense layer(s). The function should return a
          tf.Tensor.
      output_b_init (callable): Initializer function for the bias
          of output dense layer(s). The function should return a
          tf.Tensor.
      hidden_state_init (callable): Initializer function for the
          initial hidden state. The functino should return a tf.Tensor.
      hidden_state_init_trainable (bool): Bool for whether the initial
          hidden state is trainable.
      cell_state_init (callable): Initializer function for the
          initial cell state. The functino should return a tf.Tensor.
      cell_state_init_trainable (bool): Bool for whether the initial
          cell state is trainable.
      forget_bias (bool): If True, add 1 to the bias of the forget gate at
          initialization. It's used to reduce the scale of forgetting at the
          beginning of the training.
      layer_normalization (bool): Bool for using layer normalization or not.
class_docstr_tok: ["LSTM","Model",".","Args",":","output_dim","(","int",")",":","Dimension","of","the","network","output",".","hidden_dim","(","int",")",":","Hidden","dimension","for","LSTM","cell",".","name","(","str",")",":","Policy","name",",","also","the","variable","scope",".","hidden_nonlinearity","(","callable",")",":","Activation","function","for","intermediate","dense","layer","(","s",")",".","It","should","return","a","tf.Tensor",".","Set","it","to","None","to","maintain","a","linear","activation",".","hidden_w_init","(","callable",")",":","Initializer","function","for","the","weight","of","intermediate","dense","layer","(","s",")",".","The","function","should","return","a","tf.Tensor",".","hidden_b_init","(","callable",")",":","Initializer","function","for","the","bias","of","intermediate","dense","layer","(","s",")",".","The","function","should","return","a","tf.Tensor",".","recurrent_nonlinearity","(","callable",")",":","Activation","function","for","recurrent","layers",".","It","should","return","a","tf.Tensor",".","Set","it","to","None","to","maintain","a","linear","activation",".","recurrent_w_init","(","callable",")",":","Initializer","function","for","the","weight","of","recurrent","layer","(","s",")",".","The","function","should","return","a","tf.Tensor",".","output_nonlinearity","(","callable",")",":","Activation","function","for","output","dense","layer",".","It","should","return","a","tf.Tensor",".","Set","it","to","None","to","maintain","a","linear","activation",".","output_w_init","(","callable",")",":","Initializer","function","for","the","weight","of","output","dense","layer","(","s",")",".","The","function","should","return","a","tf.Tensor",".","output_b_init","(","callable",")",":","Initializer","function","for","the","bias","of","output","dense","layer","(","s",")",".","The","function","should","return","a","tf.Tensor",".","hidden_state_init","(","callable",")",":","Initializer","function","for","the","initial","hidden","state",".","The","functino","should","return","a","tf.Tensor",".","hidden_state_init_trainable","(","bool",")",":","Bool","for","whether","the","initial","hidden","state","is","trainable",".","cell_state_init","(","callable",")",":","Initializer","function","for","the","initial","cell","state",".","The","functino","should","return","a","tf.Tensor",".","cell_state_init_trainable","(","bool",")",":","Bool","for","whether","the","initial","cell","state","is","trainable",".","forget_bias","(","bool",")",":","If","True",",","add","1","to","the","bias","of","the","forget","gate","at","initialization",".","It","'s","used","to","reduce","the","scale","of","forgetting","at","the","beginning","of","the","training",".","layer_normalization","(","bool",")",":","Bool","for","using","layer","normalization","or","not","."]
code: def __init__(
      self,
      output_dim,
      hidden_dim,
      name=None,
      hidden_nonlinearity=tf.nn.tanh,
      hidden_w_init=tf.initializers.glorot_uniform(
          seed=deterministic.get_tf_seed_stream()
      ),
      hidden_b_init=tf.zeros_initializer(),
      recurrent_nonlinearity=tf.nn.sigmoid,
      recurrent_w_init=tf.initializers.glorot_uniform(
          seed=deterministic.get_tf_seed_stream()
      ),
      output_nonlinearity=None,
      output_w_init=tf.initializers.glorot_uniform(
          seed=deterministic.get_tf_seed_stream()
      ),
      output_b_init=tf.zeros_initializer(),
      hidden_state_init=tf.zeros_initializer(),
      hidden_state_init_trainable=False,
      cell_state_init=tf.zeros_initializer(),
      cell_state_init_trainable=False,
      forget_bias=True,
      layer_normalization=False,
  ):
      super().__init__(name)
      self._output_dim = output_dim
      self._hidden_dim = hidden_dim
      self._forget_bias = forget_bias
      self._hidden_nonlinearity = hidden_nonlinearity
      self._hidden_w_init = hidden_w_init
      self._hidden_b_init = hidden_b_init
      self._recurrent_nonlinearity = recurrent_nonlinearity
      self._recurrent_w_init = recurrent_w_init
      self._hidden_state_init = hidden_state_init
      self._hidden_state_init_trainable = hidden_state_init_trainable
      self._cell_state_init = cell_state_init
      self._cell_state_init_trainable = cell_state_init_trainable
      self._output_nonlinearity = output_nonlinearity
      self._output_w_init = output_w_init
      self._output_b_init = output_b_init
      self._layer_normalization = layer_normalization
      self._initialize()
  
file_path: garage/src/garage/tf/models/lstm_model.py
imports: import tensorflow
  from garage.experiment import deterministic
  from garage.tf.models.lstm import lstm
  from garage.tf.models.model import Model
label_desc: Use image node_id 1 to create a new LSTMModel object from inherited base classes: Model with example: obj = LSTMModel(output_dim, hidden_dim, name, hidden_nonlinearity, hidden_w_init, hidden_b_init, recurrent_nonlinearity, recurrent_w_init, output_nonlinearity, output_w_init, output_b_init, hidden_state_init, hidden_state_init_trainable, cell_state_init, cell_state_init_trainable, forget_bias, layer_normalization)