tfa.text.crf_decode_backward

Computes backward decoding in a linear-chain CRF.

View aliases

Main aliases

tfa.text.crf.crf_decode_backward

tfa.text.crf_decode_backward(
    inputs: tfa.types.TensorLike,
    state: tfa.types.TensorLike
) -> tf.Tensor

Args
inputs	A [batch_size, num_tags] matrix of backpointer of next step (in time order).
state	A [batch_size, 1] matrix of tag index of next step.

Returns
new_tags	A [batch_size, num_tags] tensor containing the new tag indices.