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

Python tf.random.experimental.stateless_split用法及代碼示例


通過添加引導軸將 RNG 種子拆分為 num 新種子。

用法

tf.random.experimental.stateless_split(
    seed, num=2, alg='auto_select'
)

參數

  • seed 一個 RNG 種子(一個具有形狀 [2] 和 dtype int32 或 int64 的張量)。 (使用 XLA 時，僅允許使用 int32。)
  • num 可選，正整數或標量張量，指示要產生的種子數(默認為 2)。
  • alg 用於生成隨機數的 RNG 算法。有關詳細說明，請參閱tf.random.stateless_uniform。

返回

  • 形狀為 [num, 2] 的張量表示 num 新種子。它將具有與 seed 相同的 dtype(如果 seed 沒有明確的 dtype，則 dtype 將由 tf.convert_to_tensor 確定)。

例子：

seed = [1, 2]
new_seeds = tf.random.experimental.stateless_split(seed, num=3)
print(new_seeds)
tf.Tensor(
[[1105988140 1738052849]
 [-335576002  370444179]
 [  10670227 -246211131]], shape=(3, 2), dtype=int32)
tf.random.stateless_normal(shape=[3], seed=new_seeds[0,:])
<tf.Tensor:shape=(3,), dtype=float32, numpy=array([-0.59835213, -0.9578608 ,
0.9002807 ], dtype=float32)>

相關用法


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