Description Usage Arguments Value
Base abstract class that allows the user to customize sampling.
1 2 3 4 5 6 7 | sampler_custom(
initialize_fn,
sample_fn,
next_inputs_fn,
sample_ids_shape = NULL,
sample_ids_dtype = NULL
)
|
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. |
None
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.