code: def __init__(
      self,
      input_size,
      hidden_size,
      forget_bias,  # Currently unused!  Values here will be ignored.
      memory_optimization,
      drop_states=False,
      linear_before_reset=False,
      **kwargs,
  ):
      super().__init__(**kwargs)
      self.input_size = input_size
      self.hidden_size = hidden_size
      self.forget_bias = float(forget_bias)
      self.memory_optimization = memory_optimization
      self.drop_states = drop_states
      self.linear_before_reset = linear_before_reset
  
file_path: pytorch/caffe2/python/gru_cell.py
imports: import functools
  from caffe2.python import brew, rnn_cell
label_desc: Use image node_id 1 to create a new GRUCell object from inherited base classes: rnn_cell with example: obj = GRUCell(input_size, hidden_size, forget_bias, memory_optimization, drop_states, linear_before_reset)