tflittle_gru.py 文件源码

def __init__(self,n_classes,rnn_size = 256):
        self._name = "little_gru"
        self._layer_weights = tf.Variable(tf.random_uniform([rnn_size,n_classes]), name="weights")
        self._layer_biases = tf.Variable(tf.random_uniform([n_classes]), name="biases")
        self._gru_cell = rnn.GRUCell(rnn_size)