sampler_greedy_embedding: Greedy Embedding Sampler

Description Usage Arguments Details Value

Description

A sampler for use during inference.

Usage

1

Arguments

embedding_fn

A optional callable that takes a vector tensor of ids (argmax ids), or the params argument for embedding_lookup. The returned tensor will be passed to the decoder input. Default to use tf$nn$embedding_lookup.

Details

Uses the argmax of the output (treated as logits) and passes the result through an embedding layer to get the next input.

Value

None


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