torch.as_strided_scatter

torch.as_strided_scatter(input, src, size, stride, storage_offset=None) → Tensor

Встраивает значения тензора src в input по элементам, соответствующим результату вызова input.as_strided(size, stride, storage_offset).

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

Параметры:

• input (Tensor) – входной тензор.
• size (tuple или ints) – форма выходного тензора
• stride (tuple или ints) – шаг выходного тензора
• storage_offset (int, optional) – смещение в базовом хранилище выходного тензора

Примечание

src должен иметь подходящий размер для встраивания в input. В частности, он должен иметь ту же форму, что и torch.as_strided(input, size, stride, storage_offset)

Пример:

>>> a = torch.arange(4).reshape(2, 2) + 1
>>> a
tensor([[1, 2],
        [3, 4]])
>>> b = torch.zeros(3, 3)
>>> b
tensor([[0., 0., 0.],
        [0., 0., 0.],
        [0., 0., 0.]])
>>> torch.as_strided_scatter(b, a, (2, 2), (1, 2))
tensor([[1., 3., 2.],
        [4., 0., 0.],
        [0., 0., 0.]])