deep_gru_lstm_init_gru.py

def _init_gru(self):
    
    for s in ['wr', 'wu', 'wa']:
        setattr(self, s, [tf.Variable(tf.random.normal(
                                stddev=std_dev,
                                shape=shape,
                                dtype=tf.double))
                            for shape, std_dev in self.w_shapes])
    
    for s in ['br', 'bu', 'ba']:
        setattr(self, s, [tf.Variable(tf.random.normal(
                                stddev=std_dev,
                                shape=shape,
                                dtype=tf.double))
                            for shape, std_dev in self.b_shapes])
    
    all_weights = []
    for w in [self.wr, self.br, self.wu, self.bu, self.wa, self.ba]:
        all_weights.extend(w)
    
    return all_weights