tf.raw_ops.StridedSlice

Возвращает срез input с шагом.

tf.raw_ops.StridedSlice(
    input,
    begin,
    end,
    strides,
    begin_mask=0,
    end_mask=0,
    ellipsis_mask=0,
    new_axis_mask=0,
    shrink_axis_mask=0,
    name=None
)

Обратите внимание, что большинство пользователей Python захотят использовать Python Tensor.getitem или Variable.getitem, а не этот оператор напрямую.

Цель этого оператора — создать новый тензор с подмножеством элементов из n-мерного input тензора. Подмножество выбирается с помощью последовательности m спецификаций разряженных диапазонов, закодированных в аргументах этой функции. Обратите внимание, что в некоторых случаях m может быть равно n, но это необязательно. Каждый элемент спецификации диапазона может быть одним из следующих:

• Эллипсис (...). Эллипсы используются для подразумевания нуля или более измерений полного выбора размеров и создаются с помощью ellipsis_mask. Например, foo[...] — это срез тождества.

• Новая ось. Это используется для вставки нового измерения с размером 1, создается с помощью new_axis_mask. Например, foo[:, ...] где foo имеет размер (3, 4) создает (1, 3, 4) тензор.

• Диапазон begin:end:stride. Это используется для указания количества выбираемых элементов из данного измерения. stride может быть любым целым числом, но не 0. begin — это целое число, которое представляет индекс первого выбираемого значения, а end — индекс последнего выбираемого значения. Количество выбранных значений в каждом измерении равно end - begin если stride > 0 и begin - end если stride < 0. begin и end могут быть отрицательными, где -1 — последний элемент, -2 — предпоследний. begin_mask определяет, следует ли заменить явно заданное begin неявным эффективным значением 0 если stride > 0 или -1 если stride < 0. end_mask аналогичен, но вычисляет количество, необходимое для создания наибольшего открытого интервала. Например, для тензора формы (3,) foo[:], эффективные значения begin и end составляют 0 и 3. Не предполагайте, что это эквивалентно foo[0:-1], которое имеет эффективные значения begin и end равные 0 и 2. Еще один пример — foo[-2::-1], который инвертирует первое измерение тензора, опускает последние два (в исходном порядке элементов). Например, foo = [1,2,3,4]; foo[-2::-1] равно [4,3].

• Один индекс. Используется для сохранения только элементов с заданным индексом. Например, (foo[2, :] на тензоре формы (5,6) создает тензор формы (6,) . Это закодировано в begin и end и shrink_axis_mask.

Каждая концептуальная спецификация диапазона закодирована в аргументе оператора. Лучше всего понять эту кодировку, рассмотрев нетривиальный пример. В частности, foo[1, 2:4, None, ..., :-3:-1, :] будет закодирован как

begin = [1, 2, x, x, 0, x] # x denotes don't care (usually 0)
end = [2, 4, x, x, -3, x]
strides = [1, 1, x, x, -1, 1]
begin_mask = 1<<4 | 1<<5 = 48
end_mask = 1<<5 = 32
ellipsis_mask = 1<<3 = 8
new_axis_mask = 1<<2 = 4
shrink_axis_mask = 1<<0 = 1

В этом случае, если foo.shape равно (5, 5, 5, 5, 5, 5), конечная форма среза станет (2, 1, 5, 5, 2, 5). Давайте шаг за шагом рассмотрим каждую спецификацию аргумента.

1. Первый аргумент в примере среза преобразуется в begin = 1 и end = begin + 1 = 2. Для того, чтобы отличить от исходной спецификации 2:4, мы также устанавливаем соответствующий бит в shrink_axis_mask.

2. 2:4 вносит 2, 4, 1 в начало, конец и шаг. Все маски имеют нулевые биты.

3. None — синоним для tf.newaxis. Это означает вставку измерения размером 1 в конечную форму. В начале, конце и шаге вносятся фиктивные значения, а бит new_axis_mask устанавливается.

4. ... берет полные диапазоны из необходимого количества измерений, чтобы полностью указать срез для каждого измерения формы входного тензора.

5. :-3:-1 демонстрирует использование отрицательных индексов. Отрицательный индекс i в измерении с формой s преобразуется в положительный индекс s + i. Таким образом, -1 становится s-1 (т.е. последний элемент). Это преобразование выполняется внутри, поэтому начало, конец и шаги получают значения x, -3 и -1 соответственно. Соответствующий бит begin_mask устанавливается, чтобы указать, что начальный диапазон — это полный диапазон (без учёта x).

6. : указывает, что выбираются все элементы соответствующего измерения. Это эквивалентно :: или 0::1. Начало, конец и шаги получают значения 0, 0 и 1 соответственно. Соответствующие биты в begin_mask и end_mask также устанавливаются.

Требования: 0 != strides[i] for i in [0, m) ellipsis_mask must be a power of two (only one ellipsis)