View source on GitHub |
Constrains the inputs to filter out certain tags at each time step.
tfa.text.crf_filtered_inputs(
inputs: tfa.types.TensorLike
,
tag_bitmap: tfa.types.TensorLike
) -> tf.Tensor
tag_bitmap limits the allowed tags at each input time step. This is useful when an observed output at a given time step needs to be constrained to a selected set of tags.
Returns | |
---|---|
filtered_inputs
|
A [batch_size] vector of unnormalized sequence scores. |