Draws samples from a multinomial distribution.
tf.raw_ops.Multinomial( logits, num_samples, seed=0, seed2=0, output_dtype=tf.dtypes.int64, name=None )
Args | |
---|---|
logits | A Tensor . Must be one of the following types: float32 , float64 , int32 , uint8 , int16 , int8 , int64 , bfloat16 , uint16 , half , uint32 , uint64 . 2-D Tensor with shape [batch_size, num_classes] . Each slice [i, :] represents the unnormalized log probabilities for all classes. |
num_samples | A Tensor of type int32 . 0-D. Number of independent samples to draw for each row slice. |
seed | An optional int . Defaults to 0 . If either seed or seed2 is set to be non-zero, the internal random number generator is seeded by the given seed. Otherwise, a random seed is used. |
seed2 | An optional int . Defaults to 0 . A second seed to avoid seed collision. |
output_dtype | An optional tf.DType from: tf.int32, tf.int64 . Defaults to tf.int64 . |
name | A name for the operation (optional). |
Returns | |
---|---|
A Tensor of type output_dtype . |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/raw_ops/Multinomial