Computes forward decoding in a linear-chain CRF.
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.
|