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