TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

tf.contrib.crf.crf_unary_score

View source on GitHub

Computes the unary scores of tag sequences.

tf.contrib.crf.crf_unary_score(
    tag_indices,
    sequence_lengths,
    inputs
)

Args:

  • tag_indices: A [batch_size, max_seq_len] matrix of tag indices.
  • sequence_lengths: A [batch_size] vector of true sequence lengths.
  • inputs: A [batch_size, max_seq_len, num_tags] tensor of unary potentials.

Returns:

  • unary_scores: A [batch_size] vector of unary scores.