tf.random.stateless_binomial
Выводит детерминированные псевдослучайные значения из биномиального распределения.
tf.random.stateless_binomial(
shape,
seed,
counts,
probs,
output_dtype=tf.dtypes.int32,
name=None
)
Сгенерированные значения следуют биномиальному распределению со значениями параметров количества испытаний и вероятности успеха.
Это бессостоятельное (stateless) исполнение tf.random.Generator.binomial: если запустить его дважды с одинаковыми значениями семян и формами, он сгенерирует одни и те же псевдослучайные числа. Выходной результат будет одинаковым при многократных запусках на одном оборудовании (и между CPU и GPU), но может меняться между версиями TensorFlow или на оборудовании, отличном от CPU/GPU.
Пример:
counts = [10., 20.]
# Probability of success.
probs = [0.8]
binomial_samples = tf.random.stateless_binomial(
shape=[2], seed=[123, 456], counts=counts, probs=probs)
counts = ... # Shape [3, 1, 2]
probs = ... # Shape [1, 4, 2]
shape = [3, 4, 3, 4, 2]
# Sample shape will be [3, 4, 3, 4, 2]
binomial_samples = tf.random.stateless_binomial(
shape=shape, seed=[123, 456], counts=counts, probs=probs)
| Аргументы | |
|---|---|
shape | 1-мерный целочисленный тензор или Python-массив. Форма выходного тензора. |
seed | Тензор формы [2], семя для генератора псевдослучайных чисел. Должен иметь тип int32 или int64. (При использовании XLA разрешен только тип int32.) |
counts | Тензор. Количество испытаний в биномиальном распределении. Должен быть совместим по трансляции с probs, и совместим по трансляции с правыми измерениями shape. |
probs | Тензор. Вероятность успеха в биномиальном распределении. Должен быть совместим по трансляции с counts и совместим по трансляции с правыми измерениями shape. |
output_dtype | Тип выходного значения. По умолчанию: tf.int32 |
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
samples | Тензор заданной формы, заполненный случайными биномиальными значениями. Для каждого i, каждый элемент samples[..., i] — независимое значение из биномиального распределения на counts[i] испытаниях с вероятностью успеха probs[i]. |
© 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/stateless_binomial