sampler_greedy_embedding {tfaddons}

Greedy Embedding Sampler

Description

A sampler for use during inference.

Usage

sampler_greedy_embedding(embedding_fn = NULL)

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