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


Python tf.compat.v1.random.stateless_multinomial用法及代碼示例


從多項分布中抽取確定性偽隨機樣本。 (已棄用)

用法

tf.compat.v1.random.stateless_multinomial(
    logits, num_samples, seed, output_dtype=tf.dtypes.int64, name=None
)

參數

  • logits 形狀為 [batch_size, num_classes] 的二維張量。每個切片 [i,:] 代表所有類的非規範化 log-probabilities。
  • num_samples 0-D。為每個行切片繪製的獨立樣本數。
  • seed 形狀 [2] 張量,隨機數生成器的種子。必須具有數據類型 int32int64 。 (使用 XLA 時,僅允許使用 int32。)
  • output_dtype 輸出的整數類型:int32int64。默認為 int64
  • name 操作的可選名稱。

返回

  • 形狀 [batch_size, num_samples] 的繪製樣本。

警告:此函數已棄用。它將在未來的版本中刪除。更新說明:改用tf.random.stateless_categorical

這是tf.random.categorical 的無狀態版本:如果使用相同的種子和形狀運行兩次,它將產生相同的偽隨機數。輸出在同一硬件(以及 CPU 和 GPU 之間)上的多次運行中是一致的,但可能會在 TensorFlow 版本之間或在非 CPU/GPU 硬件上發生變化。

例子:

# samples has shape [1, 5], where each value is either 0 or 1 with equal
# probability.
samples = tf.random.stateless_categorical(
    tf.math.log([[0.5, 0.5]]), 5, seed=[7, 17])

相關用法


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