tf.random.categorical
Генерирует выборки из категориального распределения.
tf.random.categorical(
logits, num_samples, dtype=None, seed=None, name=None
)
Пример:
# 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 | 2-мерный тензор с формой [batch_size, num_classes]. Каждый срез [i, :] представляет ненормализованные логарифмические вероятности для всех классов. |
num_samples | 0-мерный. Количество независимых выборок для каждого среза строки. |
dtype | Тип целого числа для использования в выводе. По умолчанию int64. |
seed | Целое число Python. Используется для создания случайного зерна для распределения. См. tf.compat.v1.set_random_seed для поведения. |
name | Необязательное имя для операции. |
| Возвращаемое значение |
|---|
Выбранные образцы формы [batch_size, num_samples]. |
