tf.keras.random.categorical | TensorFlow v2.16.1 (original) (raw)
tf.keras.random.categorical
Stay organized with collections Save and categorize content based on your preferences.
Draws samples from a categorical distribution.
tf.keras.random.categorical(
logits, num_samples, dtype='int32', seed=None
)
This function takes as input logits, a 2-D input tensor with shape (batch_size, num_classes). Each row of the input represents a categorical distribution, with each column index containing the log-probability for a given class.
The function will output a 2-D tensor with shape (batch_size, num_samples), where each row contains samples from the corresponding row in logits. Each column index contains an independent samples drawn from the input distribution.
| Args | |
|---|---|
| logits | 2-D Tensor with shape (batch_size, num_classes). Each row should define a categorical distibution with the unnormalized log-probabilities for all classes. |
| num_samples | Int, the number of independent samples to draw for each row of the input. This will be the second dimension of the output tensor's shape. |
| dtype | Optional dtype of the output tensor. |
| seed | A Python integer or instance ofkeras.random.SeedGenerator. Used to make the behavior of the initializer deterministic. Note that an initializer seeded with an integer or None (unseeded) will produce the same random values across multiple calls. To get different random values across multiple calls, use as seed an instance of keras.random.SeedGenerator. |
| Returns |
|---|
| A 2-D tensor with (batch_size, num_samples). |
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.
Last updated 2024-06-07 UTC.