tf.raw_ops.Roll
Поворачивает элементы тензора вдоль оси.
tf.raw_ops.Roll(
input, shift, axis, name=None
)
Элементы сдвигаются положительно (в сторону больших индексов) на смещение shift вдоль размерности axis. Отрицательные значения shift сместят элементы в противоположном направлении. Элементы, которые прошли за последнюю позицию, перейдут в первую и наоборот. Можно указать несколько сдвигов по нескольким осям.
Например:
# 't' is [0, 1, 2, 3, 4] roll(t, shift=2, axis=0) ==> [3, 4, 0, 1, 2] # shifting along multiple dimensions # 't' is [[0, 1, 2, 3, 4], [5, 6, 7, 8, 9]] roll(t, shift=[1, -2], axis=[0, 1]) ==> [[7, 8, 9, 5, 6], [2, 3, 4, 0, 1]] # shifting along the same axis multiple times # 't' is [[0, 1, 2, 3, 4], [5, 6, 7, 8, 9]] roll(t, shift=[2, -3], axis=[1, 1]) ==> [[1, 2, 3, 4, 0], [6, 7, 8, 9, 5]]
| Аргументы | |
|---|---|
input | A Tensor. |
shift | A Tensor. Должен быть одного из следующих типов: int32, int64. Размерность должна быть 0-D или 1-D. shift[i] указывает количество позиций, на которые элементы сдвигаются положительно (в сторону больших индексов) вдоль размерности, указанной в axis[i]. Отрицательные сдвиги будут поворачивать элементы в противоположном направлении. |
axis | A Tensor. Должен быть одного из следующих типов: int32, int64. Размерность должна быть 0-D или 1-D. axis[i] указывает размерность, по которой должен произойти сдвиг shift[i]. Если одна и та же ось упоминается более одного раза, то общий сдвиг для этой оси будет суммой всех сдвигов, относящихся к этой оси. |
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
A Tensor. Имеет тот же тип, что и input. |
© 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/Roll