tfa.text.crf_decode_backward

View source on GitHub

Computes backward decoding in a linear-chain CRF.

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.

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