Samples from categorical distribution.

sample_categorical(
  logits,
  dtype = tf$int32,
  sample_shape = list(),
  seed = NULL
)

Arguments

logits

logits

dtype

dtype

sample_shape

the shape of sample

seed

random seed: integer

Value

a Tensor