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 имеют одинаковую последовательность начальных измерений, а обновления выполняются по последнему измерению indices. Иными словами, измерения должны иметь следующий вид:

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.

Эту операцию можно заменить двумя способами:

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.