Warning: This project is deprecated. TensorFlow Addons has stopped development, The project will only be providing minimal maintenance releases until May 2024. See the full announcement here or on github.


Constrains the inputs to filter out certain tags at each time step.

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.

inputs A [batch_size, max_seq_len, num_tags] tensor of unary potentials to use as input to the CRF layer.
tag_bitmap A [batch_size, max_seq_len, num_tags] boolean tensor representing all active tags at each index for which to calculate the unnormalized score.

filtered_inputs A [batch_size] vector of unnormalized sequence scores.