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