crf_decode_forward {tfaddons}

CRF decode forward

Description

Computes forward decoding in a linear-chain CRF.

Usage

crf_decode_forward(inputs, state, transition_params, sequence_lengths)

Arguments

inputs

A [batch_size, num_tags] matrix of unary potentials.

state

A [batch_size, num_tags] matrix containing the previous step's score values.

transition_params

A [num_tags, num_tags] matrix of binary potentials.

sequence_lengths

A [batch_size] vector of true sequence lengths.

Value

backpointers: A [batch_size, num_tags] matrix of backpointers. new_state: A [batch_size, num_tags] matrix of new score values.