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 (tuple of LongTensor) – тензоры, используемые для индексации self.
• values (Tensor) – тензор того же типа данных, что и self.
• accumulate (bool) – указывает, следует ли накапливать значения в self