tfa.text.crf_decode_forward

View source on GitHub

Computes forward decoding in a linear-chain CRF.

tfa.text.crf_decode_forward(
    inputs,
    state,
    transition_params,
    sequence_lengths
)

Args:

• inputs: A [batch_size, num_tags] matrix of unary potentials.
• state: A [batch_size, num_tags] matrix containing the previous step's score values.
• transition_params: A [num_tags, num_tags] matrix of binary potentials.
• sequence_lengths: A [batch_size] vector of true sequence lengths.

Returns:

• backpointers: A [batch_size, num_tags] matrix of backpointers.
• new_state: A [batch_size, num_tags] matrix of new score values.