tfa.text.crf_log_norm

Computes the normalization for a CRF.

View aliases

Main aliases

tfa.text.crf.crf_log_norm

tfa.text.crf_log_norm(
    inputs: tfa.types.TensorLike,
    sequence_lengths: tfa.types.TensorLike,
    transition_params: tfa.types.TensorLike
) -> tf.Tensor

Args
inputs	A [batch_size, max_seq_len, num_tags] tensor of unary potentials to use as input to the CRF layer.
sequence_lengths	A [batch_size] vector of true sequence lengths.
transition_params	A [num_tags, num_tags] transition matrix.

Returns
log_norm	A [batch_size] vector of normalizers for a CRF.