sampler_custom {tfaddons}

Base abstract class that allows the user to customize sampling.

Description

Base abstract class that allows the user to customize sampling.

Usage

sampler_custom(
  initialize_fn,
  sample_fn,
  next_inputs_fn,
  sample_ids_shape = NULL,
  sample_ids_dtype = NULL
)

Arguments

initialize_fn

callable that returns (finished, next_inputs) for the first iteration.

sample_fn

callable that takes (time, outputs, state) and emits tensor sample_ids.

next_inputs_fn

callable that takes (time, outputs, state, sample_ids) and emits (finished, next_inputs, next_state).

sample_ids_shape

Either a list of integers, or a 1-D Tensor of type int32, the shape of each value in the sample_ids batch. Defaults to a scalar.

sample_ids_dtype

The dtype of the sample_ids tensor. Defaults to int32.

Value

None