crf_log_norm: CRF log norm

Description Usage Arguments Value

View source: R/text_ops.R

Description

Computes the normalization for a CRF.

Usage

1
crf_log_norm(inputs, 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.

sequence_lengths

A [batch_size] vector of true sequence lengths.

transition_params

A [num_tags, num_tags] transition matrix.

Value

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


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