torch.diagonal_scatter

torch.diagonal_scatter(input, src, offset=0, dim1=0, dim2=1) → Tensor

Встраивает значения тензора src в input по диагональным элементам input, относительно dim1 и dim2.

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

Аргумент offset управляет тем, какую диагональ учитывать:

• Если offset = 0, это главная диагональ.
• Если offset > 0, это диагональ выше главной.
• Если offset < 0, это диагональ ниже главной.

Параметры

• input (Тензор) – входной тензор. Должен быть по крайней мере двумерным.
• src (Тензор) – тензор, который необходимо встроить в input.
• offset (int, необязательно) – какая диагональ должна быть рассмотрена. По умолчанию: 0 (главная диагональ).
• dim1 (int, необязательно) – первый размер, относительно которого нужно взять диагональ. По умолчанию: 0.
• dim2 (int, необязательно) – второй размер, относительно которого нужно взять диагональ. По умолчанию: 1.

Примечание

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

Примеры:

>>> a = torch.zeros(3, 3)
>>> a
tensor([[0., 0., 0.],
        [0., 0., 0.],
        [0., 0., 0.]])

>>> torch.diagonal_scatter(a, torch.ones(3), 0)
tensor([[1., 0., 0.],
        [0., 1., 0.],
        [0., 0., 1.]])

>>> torch.diagonal_scatter(a, torch.ones(2), 1)
tensor([[0., 1., 0.],
        [0., 0., 1.],
        [0., 0., 0.]])