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)