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 | Тензор Tensor. Тензор, который нужно перемешать. |
key | Тензор Tensor типа uint64. Ключ для алгоритма ГСЧ на основе счётчика (форма uint64[1]). |
counter | Тензор Tensor типа uint64. Начальное значение счётчика для алгоритма ГСЧ на основе счётчика (форма uint64[2] или uint64[1] в зависимости от алгоритма). Если задан вектор большего размера, будет использована только необходимая часть слева (то есть [:N]). |
alg | Тензор Tensor типа int32. Алгоритм ГСЧ (форма int32[]). |
name | Имя операции (необязательно). |
| Возвращает | |
|---|---|
Тензор 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