import tensorflow as tf

  #     inputs=embedding_output, 
  #     sequence_lengths=_true_length, 
  #     rnn_size=config.lstm_size,
  #     dropout=config.bilstm_dropout_rate, 
  #     is_training=is_training,
  #     num_layers=config.num_bilstm,
  #     direction='bidirectional')

  # first_token_tensor = tf.squeeze(sequence_output[:, -1:, :], axis=1)
  last_token_tensor = tf.squeeze(sequence_output[:, -1:, :], axis=1)
  output_layer = tf.layers.dense(
            last_token_tensor,
            config.hidden_size,
            activation=tf.tanh,
            kernel_initializer=modeling.create_initializer(config.initializer_range))

  hidden_size = output_layer.shape[-1].value

