tf.strided_slice

Извлекает строенный срез тензора (обобщенная индексация массивов Python).

tf.strided_slice(
    input_, begin, end, strides=None, begin_mask=0, end_mask=0, ellipsis_mask=0,
    new_axis_mask=0, shrink_axis_mask=0, var=None, name=None
)

См. также tf.slice.

Вместо прямого вызова этого оператора большинство пользователей будут использовать синтаксис срезов в стиле NumPy (например, tensor[..., 3:4:-1, tf.newaxis, 3]), который поддерживается с помощью tf.Tensor.getitem и tf.Variable.getitem. Интерфейс этого оператора — это низкоуровневое кодирование синтаксиса срезов.

Грубо говоря, этот оператор извлекает срез размером (end-begin)/stride из данного input_ тензора. Начиная с позиции, указанной begin, срез продолжается путем добавления stride к индексу до тех пор, пока все размеры не станут не меньше end. Обратите внимание, что шаг может быть отрицательным, что приводит к обращенному срезу.

Учитывая срез Python input[spec0, spec1, ..., specn], эта функция будет вызвана следующим образом.

begin, end, и strides будут векторами длиной n. n в общем случае не равен рангу input_ тензора.

В каждом поле маски (begin_mask, end_mask, ellipsis_mask, new_axis_mask, shrink_axis_mask) i-й бит будет соответствовать i-й спецификации.

Если i-й бит begin_mask установлен, то begin[i] игнорируется, и вместо этого используется максимально возможный диапазон в этом измерении. end_mask работает аналогично, но с конечным диапазоном.

foo[5:,:,:3] над тензором 7x8x9 эквивалентно foo[5:7,0:8,0:3]. foo[::-1] инвертирует тензор с формой 8.

Если i-й бит ellipsis_mask установлен, то столько не указанных измерений, сколько необходимо, будет вставлено между другими измерениями. В ellipsis_mask разрешен только один ненулевой бит.

Например, foo[3:5,...,4:5] над тензором формы 10x3x3x10 эквивалентно foo[3:5,:,:,4:5], и foo[3:5,...] эквивалентно foo[3:5,:,:,:].

Если i-й бит new_axis_mask установлен, то begin, end, и stride игнорируются, и к этому месту в выходном тензоре добавляется новое измерение длиной 1.

Например, foo[:4, tf.newaxis, :2] создаст тензор формы (4, 1, 2).

Если i-й бит shrink_axis_mask установлен, это означает, что i-я спецификация уменьшает размерность на 1, принимая значение по индексу begin[i]. end[i] и strides[i] игнорируются в этом случае. Например, в Python можно сделать foo[:, 3, :], что приведет к shrink_axis_mask равному 2.

Примечание: begin и end нулевые индексации. strides записи должны быть отличны от нуля.

t = tf.constant([[[1, 1, 1], [2, 2, 2]],
                 [[3, 3, 3], [4, 4, 4]],
                 [[5, 5, 5], [6, 6, 6]]])
tf.strided_slice(t, [1, 0, 0], [2, 1, 3], [1, 1, 1])  # [[[3, 3, 3]]]
tf.strided_slice(t, [1, 0, 0], [2, 2, 3], [1, 1, 1])  # [[[3, 3, 3],
                                                      #   [4, 4, 4]]]
tf.strided_slice(t, [1, -1, 0], [2, -3, 3], [1, -1, 1])  # [[[4, 4, 4],
                                                         #   [3, 3, 3]]]