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 | Целочисленный тип результата: int32 или int64. По умолчанию int64. |
seed | Целое число Python. Используется для создания случайного семени для распределения. См. tf.random.set_seed для получения информации о поведении. |
name | Необязательное имя операции. |
| Возвращает |
|---|
Выбранные образцы с формой [batch_size, num_samples]. |
