tf.keras.ops.slice_update

Обновление входного значения путём срезки в тензоре обновлённых значений.

tf.keras.ops.slice_update(
    inputs, start_indices, updates
)

В общем случае, эта операция выполняет inputs[start_indices: start_indices + updates.shape] = updates. Предполагается, что входной тензор имеет размерность (D0, D1, ..., Dn), start_indices должен быть списком/кортежем из n целых чисел, определяющих начальные индексы. updates должен иметь такую же размерность, как и inputs, а размер каждой размерности не должен превышать Di - start_indices[i]. Например, если у нас есть 2D входной тензор inputs = np.zeros((5, 5)), и мы хотим обновить пересечение последних 2 строк и последних 2 столбцов значением 1, т.е. inputs[3:, 3:] = np.ones((2, 2)), то мы можем использовать следующий код:

inputs = np.zeros((5, 5))
start_indices = [3, 3]
updates = np.ones((2, 2))
inputs = keras.ops.slice_update(inputs, start_indices, updates)