View source on GitHub |
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
Returns | |
---|---|
unary_scores
|
A [batch_size] vector of unary scores. |