torch.select_scatter

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

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

Параметры

• input (Tensor) – входной тензор.
• src (Tensor) – Тензор, который нужно встроить в 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.]])