tf.keras.ops.slice

Возвращает срез входного тензора.

tf.keras.ops.slice(
    inputs, start_indices, shape
)

В общем случае, эта операция является явным заменителем среза массива, например, inputs[start_indices: start_indices + shape]. В отличие от среза с использованием квадратных скобок, эта операция будет принимать тензорные начальные индексы на всех бэкендах, что полезно, когда индексы динамически вычисляются с помощью других тензорных операций.

inputs = np.zeros((5, 5))
start_indices = np.array([3, 3])
shape = np.array([2, 2])
inputs = keras.ops.slice(inputs, start_indices, updates)