tfa.text.crf_binary_score

View source on GitHub

Computes the binary scores of tag sequences.

tag_indices A [batch_size, max_seq_len] matrix of tag indices.
sequence_lengths A [batch_size] vector of true sequence lengths.
transition_params A [num_tags, num_tags] matrix of binary potentials.

binary_scores A [batch_size] vector of binary scores.