tfa.text.crf_decode_forward

View source on GitHub

Computes forward decoding in a linear-chain CRF.

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

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.

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