tfa.text.crf_unary_score

Computes the unary scores of tag sequences.

tfa.text.crf_unary_score(
    tag_indices: tfa.types.TensorLike,
    sequence_lengths: tfa.types.TensorLike,
    inputs: tfa.types.TensorLike
) -> tf.Tensor

tag_indices A [batch_size, max_seq_len] matrix of tag indices.
sequence_lengths A [batch_size] vector of true sequence lengths.
inputs A [batch_size, max_seq_len, num_tags] tensor of unary potentials.

unary_scores A [batch_size] vector of unary scores.