tfa.text.crf_decode_forward

bookmark_border Stay organized with collections Save and categorize content based on your preferences.

View source on GitHub

Computes forward decoding in a linear-chain CRF.

View aliases

Main aliases

tfa.text.crf.crf_decode_forward

tfa.text.crf_decode_forward(
    inputs: tfa.types.TensorLike,
    state: tfa.types.TensorLike,
    transition_params: tfa.types.TensorLike,
    sequence_lengths: tfa.types.TensorLike
) -> tf.Tensor

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.