sampler_scheduled_output_training {tfaddons}

Scheduled Output Training Sampler

Description

A training sampler that adds scheduled sampling directly to outputs.

Usage

sampler_scheduled_output_training(
  sampling_probability,
  time_major = FALSE,
  seed = NULL,
  next_inputs_fn = NULL
)

Arguments

sampling_probability

A float32 scalar tensor: the probability of sampling from the outputs instead of reading directly from the inputs.

time_major

bool. Whether the tensors in inputs are time major. If False (default), they are assumed to be batch major.

seed

The sampling seed.

next_inputs_fn

(Optional) callable to apply to the RNN outputs to create the next input when sampling. If None (default), the RNN outputs will be used as the next inputs.

Value

FALSE for sample_ids where no sampling took place; TRUE elsewhere.