tf.compat.v1.batch_scatter_update

Обобщение tf.compat.v1.scatter_update на ось, отличную от 0. (устарело)

tf.compat.v1.batch_scatter_update(
    ref, indices, updates, use_locking=True, name=None
)

Аналогично batch_gather. Предполагается, что ref, indices и updates имеют серию ведущих измерений, одинаковых для всех, и обновления выполняются по последнему измерению индексов. Другими словами, измерения должны быть следующими:

num_prefix_dims = indices.ndims - 1 batch_dim = num_prefix_dims + 1 updates.shape = indices.shape + var.shape[batch_dim:]

где

updates.shape[:num_prefix_dims] == indices.shape[:num_prefix_dims] == var.shape[:num_prefix_dims]

И выполняемая операция может быть выражена как:

var[i_1, ..., i_n, indices[i_1, ..., i_n, j]] = updates[i_1, ..., i_n, j]

Когда indices является одномерным тензором, эта операция эквивалентна tf.compat.v1.scatter_update.

Чтобы избежать этой операции, есть 2 альтернативы:

1) Изменение формы переменной путём слияния первых ndims измерений. Однако это невозможно, потому что tf.reshape возвращает тензор, на который нельзя применить tf.compat.v1.scatter_update. 2) Проход по первым ndims измерениям переменной и применение tf.compat.v1.scatter_update к подтензорам, полученным путём среза первого измерения. Это действительный вариант для ndims = 1, но менее эффективен, чем это реализация.

См. также tf.compat.v1.scatter_update и tf.compat.v1.scatter_nd_update.