當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文

Python tf.random.Generator.binomial用法及代碼示例


用法

binomial(
    shape, counts, probs, dtype=tf.dtypes.int32, name=None
)

參數

  • shape 一維整數張量或 Python 數組。輸出張量的形狀。
  • counts 張量。二項分布的計數。必須可以使用 probs 進行廣播，並且可以使用 shape 的最右側維度進行廣播。
  • probs 張量。二項分布的成功概率。必須可以使用 counts 進行廣播，並且可以使用 shape 的最右側維度進行廣播。
  • dtype 輸出的類型。默認值：tf.int32
  • name 操作的名稱(可選)。

返回

  • samples 用隨機二項式值填充的指定形狀的張量。對於每個 i，每個 samples[i, ...] 都是從 counts[i] 試驗的二項式分布中獨立抽取的，具有成功概率 probs[i]。

從二項分布中輸出隨機值。

生成的值遵循具有指定計數和成功概率參數的二項分布。

例子：

counts = [10., 20.]
# Probability of success.
probs = [0.8]

rng = tf.random.Generator.from_seed(seed=234)
binomial_samples = rng.binomial(shape=[2], counts=counts, probs=probs)


counts = ... # Shape [3, 1, 2]
probs = ...  # Shape [1, 4, 2]
shape = [3, 4, 3, 4, 2]
rng = tf.random.Generator.from_seed(seed=1717)
# Sample shape will be [3, 4, 3, 4, 2]
binomial_samples = rng.binomial(shape=shape, counts=counts, probs=probs)

相關用法


注：本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.random.Generator.binomial。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。