Функция reverse

Функция reverse возвращает тензор, элементы которого расположены в обратном порядке вдоль заданных осей. Первым параметром функция принимает исходный тензор. Вторым параметром передается ось или список осей, вдоль которых нужно выполнить разворот.

Синтаксис

tf.reverse(tensor, axis)

Пример

Давайте развернем одномерный тензор 1, 2, 3, 4, 5 вдоль единственной оси:

import tensorflow as tf

t = tf.constant([1, 2, 3, 4, 5])
res = tf.reverse(t, axis=[0])
print(res)

Результат выполнения кода:

tf.Tensor([5 4 3 2 1], shape=(5,), dtype=int32)

Пример

Давайте развернем двумерный тензор только вдоль оси 0:

import tensorflow as tf

t = tf.constant([[1, 2, 3], [4, 5, 6]])
res = tf.reverse(t, axis=[0])
print(res)

Результат выполнения кода:

tf.Tensor(
[[4 5 6]
 [1 2 3]], shape=(2, 3), dtype=int32)

Пример

Давайте развернем двумерный тензор сразу вдоль обеих осей:

import tensorflow as tf

t = tf.constant([[1, 2, 3], [4, 5, 6]])
res = tf.reverse(t, axis=[0, 1])
print(res)

Результат выполнения кода:

tf.Tensor(
[[6 5 4]
 [3 2 1]], shape=(2, 3), dtype=int32)

Смотрите также

функцию transpose,
которая переставляет оси тензора
функцию strided_slice,
которая извлекает срез тензора с шагом
функцию gather,
которая собирает элементы тензора по индексам
функцию concat,
которая объединяет тензоры вдоль оси