当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


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

相关用法


注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.Tensor.fill_diagonal_。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。