torch.Tensor.index_put_

Tensor.index_put_(indices, values, accumulate=False) → Tensor

Размещает значения из тензора values в тензор self с использованием индексов, указанных в indices (который является кортежем тензоров). Выражение tensor.index_put_(indices, values) эквивалентно tensor[indices] = values. Возвращает self.

Если accumulate равно True, элементы в values добавляются к self. Если accumulate равно False, поведение неопределено, если индексы содержат дублируемые элементы.

Параметры:

• indices (кортеж LongTensor) – тензоры, используемые для индексирования self.
• values (Tensor) – тензор того же типа данных, что и self.
• accumulate (bool) – накапливать в self