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. Используется для определения детерминированного поведения инициализатора. Обратите внимание, что инициализатор, запущенный с целым числом или None (незасеянный), будет генерировать те же случайные значения при нескольких вызовах. Для получения разных случайных значений при нескольких вызовах используйте в качестве 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
