Python PyTorch Tensor.fill_diagonal_用法及代碼示例

本文簡要介紹python語言中 torch.Tensor.fill_diagonal_ 的用法。

用法:

Tensor.fill_diagonal_(fill_value, wrap=False) → Tensor

參數：

• fill_value(標量) -填充值

• wrap(bool) -高矩陣 N 列之後的對角線 ‘wrapped’。

填充至少具有 2 維的張量的主對角線。當 dims>2 時，輸入的所有維度必須等長。此函數就地修改輸入張量，並返回輸入張量。

例子：

>>> a = torch.zeros(3, 3)
>>> a.fill_diagonal_(5)
tensor([[5., 0., 0.],
        [0., 5., 0.],
        [0., 0., 5.]])
>>> b = torch.zeros(7, 3)
>>> b.fill_diagonal_(5)
tensor([[5., 0., 0.],
        [0., 5., 0.],
        [0., 0., 5.],
        [0., 0., 0.],
        [0., 0., 0.],
        [0., 0., 0.],
        [0., 0., 0.]])
>>> c = torch.zeros(7, 3)
>>> c.fill_diagonal_(5, wrap=True)
tensor([[5., 0., 0.],
        [0., 5., 0.],
        [0., 0., 5.],
        [0., 0., 0.],
        [5., 0., 0.],
        [0., 5., 0.],
        [0., 0., 5.]])