class_docstr: GRU class.
class_docstr_tok: ["GRU","class","."]
docstr: Define parameters.
docstr_tok: ["Define","parameters","."]
code: def define_params(self):
      """
      Define parameters.
      """
      input_dim = self.input_dim
      hidden_dim = self.hidden_dim
      prefix = self.name
      self.w_matrix = tf.Variable(
          tf.random_normal([input_dim, 3 * hidden_dim], stddev=0.1),
          name="/".join([prefix, "W"]),
      )
      self.U = tf.Variable(
          tf.random_normal([hidden_dim, 3 * hidden_dim], stddev=0.1),
          name="/".join([prefix, "U"]),
      )
      self.bias = tf.Variable(
          tf.random_normal([1, 3 * hidden_dim], stddev=0.1),
          name="/".join([prefix, "b"]),
      )
      return self
  
file_path: nni/examples/trials/ga_squad/rnn.py
imports: import tensorflow
  from tensorflow.python.ops.rnn_cell_impl import RNNCell
label_desc: Use image node_id 2 for calling the GRU obj's underlying member method code with example usage: obj.define_params() and returns: self