sequence.py 文件源码

def sequence_discriminator(x, length, hidden_size, scope='Discriminator', reuse=False):
  with tf.variable_scope(scope, reuse=reuse):
    x = tf.unstack(x, length, 1)
    cell = tf.nn.rnn_cell.GRUCell(hidden_size)
    _, encoding = tf.nn.rnn(x, cell)
    return layers.linear(encoding, 1), None