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.]])