torch.Tensor.index_copy_

Tensor.index_copy_(dim, index, tensor) → Tensor

Копирует элементы из tensor в тензор self, выбирая индексы в порядке, указанном в index. Например, если dim == 0 и index[i] == j, то i-я строка из tensor копируется в j-ю строку self.

Размерность dim тензора tensor должна совпадать с длиной index (которая должна быть вектором), а все остальные размерности должны совпадать с self, в противном случае будет выброшено исключение.

Примечание

Если index содержит дублирующиеся записи, несколько элементов из tensor будут скопированы в один и тот же индекс self. Результат является неопределённым, так как зависит от того, какая копия произошла последней.

Параметры

• dim (int) – размерность для индексирования
• index (LongTensor) – индексы tensor для выбора
• tensor (Tensor) – тензор, содержащий значения для копирования

Пример:

>>> x = torch.zeros(5, 3)
>>> t = torch.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]], dtype=torch.float)
>>> index = torch.tensor([0, 4, 2])
>>> x.index_copy_(0, index, t)
tensor([[ 1.,  2.,  3.],
        [ 0.,  0.,  0.],
        [ 7.,  8.,  9.],
        [ 0.,  0.,  0.],
        [ 4.,  5.,  6.]])