tf.random.experimental.stateless_split
Разделяет семя генератора случайных чисел на num новые семена, добавляя ведущую ось.
tf.random.experimental.stateless_split(
seed, num=2, alg='auto_select'
)
Пример:
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)>
| Аргументы | |
|---|---|
seed | семя генератора случайных чисел (тензор с формой [2] и типом int32 или int64). (При использовании XLA, разрешен только int32.) |
num | необязательно, положительное целое число или тензор скалярного типа, указывающий количество генерируемых семян (по умолчанию 2). |
alg | алгоритм генератора случайных чисел, используемый для генерации случайных чисел. См. tf.random.stateless_uniform для подробного объяснения. |
| Возвращаемые значения | |
|---|---|
Тензор с формой [num, 2], представляющий num новые семена. Он будет иметь тот же тип данных, что и seed (если у seed нет явного типа данных, тип данных будет определяться функцией tf.convert_to_tensor). |
© 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/versions/r2.9/api_docs/python/tf/random/experimental/stateless_split