当前位置: 首页>>代码示例 >>用法及示例精选 >>正文

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。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。