torch.select_scatter

torch.select_scatter(input, src, dim, index) → Tensor

Встраивает значения тензора src в input по заданному индексу. Эта функция возвращает тензор с новым хранилищем; она не создаёт представление.

Параметры:

• input (Тензор) – входной тензор.
• src (Тензор) – Тензор, который нужно встроить в input
• dim (int) – размерность для вставки среза.
• index (int) – индекс для выбора

Примечание

src должен иметь соответствующий размер, чтобы быть вставленным в input. В частности, он должен иметь такую же форму, что и torch.select(input, dim, index)

Пример:

>>> a = torch.zeros(2, 2)
>>> b = torch.ones(2)
>>> a.select_scatter(b, 0, 0)
tensor([[1., 1.],
        [0., 0.]])