tf.raw_ops.StatelessShuffle
Случайным и детерминированным образом перемешивает тензор по его первой размерности.
tf.raw_ops.StatelessShuffle(
value, key, counter, alg, name=None
)
Тензор перемешивается по размерности 0 таким образом, что каждый value[j] отображается на один и только один output[i]. Например, отображение, которое может произойти для тензора 3x2:
[[1, 2], [[5, 6], [3, 4], ==> [1, 2], [5, 6]] [3, 4]]
Результаты являются детерминированной функцией value, key, counter и alg.
| Аргументы | |
|---|---|
value | A Tensor. Перемешиваемый тензор. |
key | A Tensor типа uint64. Ключ для алгоритма RNG на основе счётчика (форма uint64[1]). |
counter | A Tensor типа uint64. Начальный счётчик для алгоритма RNG на основе счётчика (форма uint64[2] или uint64[1] в зависимости от алгоритма). Если предоставлен вектор большей размерности, будет использована только необходимая левая часть (т.е. [:N]). |
alg | A Tensor типа int32. Алгоритм генерации случайных чисел (форма int32[]). |
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
A Tensor. Имеет тот же тип, что и value. |
© 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/api_docs/python/tf/raw_ops/StatelessShuffle