crf_sequence_score: CRF sequence score

Description Usage Arguments Value

View source: R/text_ops.R

Description

Computes the unnormalized score for a tag sequence.

Usage

1
crf_sequence_score(inputs, tag_indices, sequence_lengths, transition_params)

Arguments

inputs

A [batch_size, max_seq_len, num_tags] tensor of unary potentials to use as input to the CRF layer.

tag_indices

A [batch_size, max_seq_len] matrix of tag indices for which we compute the unnormalized score.

sequence_lengths

A [batch_size] vector of true sequence lengths.

transition_params

A [num_tags, num_tags] transition matrix. Returns:

Value

sequence_scores: A [batch_size] vector of unnormalized sequence scores.


tfaddons documentation built on July 2, 2020, 2:12 a.m.