class_docstr: `AugmentedLSTMCell` implements a AugmentedLSTM cell.
  Args:
      embed_dim (int): The number of expected features in the input.
      lstm_dim (int): Number of features in the hidden state of the LSTM.
      Defaults to 32.
      use_highway (bool): If `True` we append a highway network to the
      outputs of the LSTM.
      use_bias (bool): If `True` we use a bias in our LSTM calculations, otherwise
      we don't.
  
  Attributes:
      input_linearity (nn.Module): Fused weight matrix which
          computes a linear function over the input.
      state_linearity (nn.Module): Fused weight matrix which
          computes a linear function over the states.
class_docstr_tok: ["`","AugmentedLSTMCell","`","implements","a","AugmentedLSTM","cell",".","Args",":","embed_dim","(","int",")",":","The","number","of","expected","features","in","the","input",".","lstm_dim","(","int",")",":","Number","of","features","in","the","hidden","state","of","the","LSTM",".","Defaults","to","32.","use_highway","(","bool",")",":","If","`","True","`","we","append","a","highway","network","to","the","outputs","of","the","LSTM",".","use_bias","(","bool",")",":","If","`","True","`","we","use","a","bias","in","our","LSTM","calculations",",","otherwise","we","do","n't",".","Attributes",":","input_linearity","(","nn.Module",")",":","Fused","weight","matrix","which","computes","a","linear","function","over","the","input",".","state_linearity","(","nn.Module",")",":","Fused","weight","matrix","which","computes","a","linear","function","over","the","states","."]
code: def reset_parameters(self):
      stdv = 1.0 / math.sqrt(self.lstm_dim)
      for weight in self.parameters():
          nn.init.uniform_(weight, -stdv, stdv)
  
imports: import math
  from typing import Optional, Tuple
  import torch
  import torch.nn
  from pytext.config import ConfigBase
  from pytext.utils.usage import log_class_usage
  from torch.nn.utils.rnn import pack_padded_sequence, PackedSequence, pad_packed_sequence
  from .representation_base import RepresentationBase
label_desc: Use image node_id 2 for calling the AugmentedLSTMCell obj's underlying member method code with example usage: obj.reset_parameters() without return types