tf.keras.random.categorical
Извлекает выборки из категориального распределения.
tf.keras.random.categorical(
logits, num_samples, dtype='int32', seed=None
)
Эта функция принимает на вход logits — двумерный входной тензор с формой (batch_size, num_classes). Каждая строка входных данных представляет категориальное распределение, а каждый индекс столбца содержит логарифм вероятности для соответствующего класса.
Функция возвращает двумерный тензор с формой (batch_size, num_samples), где каждая строка содержит выборки из соответствующей строки logits. Каждый индекс столбца содержит независимую выборку, полученную из входного распределения.
| Аргументы | |
|---|---|
logits | Двумерный тензор с формой (batch_size, num_classes). Каждая строка должна задавать категориальное распределение с ненормализованными логарифмами вероятностей для всех классов. |
num_samples | Целое число — количество независимых выборок, извлекаемых для каждой строки входных данных. Оно будет второй размерностью формы выходного тензора. |
dtype | Необязательный тип данных выходного тензора. |
seed | Целое число Python или экземпляр keras.random.SeedGenerator. Используется для обеспечения детерминированного поведения инициализатора. Обратите внимание: инициализатору, которому в качестве seed передано целое число или None (без seed), при нескольких вызовах будут соответствовать одни и те же случайные значения. Чтобы получать разные случайные значения при нескольких вызовах, используйте в качестве seed экземпляр keras.random.SeedGenerator. |
| Возвращает | |
|---|---|
| Двумерный тензор с формой (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/keras/random/categorical
