tf.random.categorical

Извлекает выборки из категориального распределения.

Совместимые псевдонимы для миграции

Подробнее см. в руководстве по миграции.

tf.random.categorical(
    logits, num_samples, dtype=None, seed=None, name=None
)

Используется в руководствах
Создание музыки с помощью RNN Игра в CartPole с помощью метода Actor-Critic Генерация текста с помощью RNN Языковые модели Wiki40B Федеративное обучение для генерации текста

# samples has shape [1, 5], where each value is either 0 or 1 with equal
# probability.
samples = tf.random.categorical(tf.math.log([[0.5, 0.5]]), 5)

Аргументы
`logits`	2D-тензор формы `[batch_size, num_classes]`. Каждый срез `[i, :]` представляет собой ненормированные логарифмы вероятностей для всех классов.
`num_samples`	0D. Количество независимых выборок, извлекаемых для каждого среза строки.
`dtype`	Целочисленный тип выходных данных: `int32` или `int64`. По умолчанию — `int64`.
`seed`	Целое число Python. Используется для создания случайного начального значения для распределения. Описание поведения см. в `tf.random.set_seed`.
`name`	Необязательное имя операции.

Возвращает
Извлечённые выборки формы `[batch_size, num_samples]`.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/random/categorical