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)