快捷方式

torch.diag

torch.diag(input, diagonal=0, *, out=None) Tensor
  • 如果 input 是一个向量(一维张量),则返回一个二维方阵张量,其中 input 的元素作为对角线。

  • 如果 input 是一个矩阵(二维张量),则返回一个一维张量,其中包含 input 的对角线元素。

参数 diagonal 控制要考虑的对角线

  • 如果 diagonal = 0,则为主对角线。

  • 如果 diagonal > 0,则位于主对角线上方。

  • 如果 diagonal < 0,则位于主对角线下方。

参数
  • input (Tensor) – 输入张量。

  • diagonal (int, 可选) – 要考虑的对角线

关键字参数

out (Tensor, 可选) – 输出张量。

另请参阅

torch.diagonal() 始终返回其输入的对角线。

torch.diagflat() 始终构造一个张量,其对角线元素由输入指定。

示例

获取输入向量作为对角线的方阵

>>> a = torch.randn(3)
>>> a
tensor([ 0.5950,-0.0872, 2.3298])
>>> torch.diag(a)
tensor([[ 0.5950, 0.0000, 0.0000],
        [ 0.0000,-0.0872, 0.0000],
        [ 0.0000, 0.0000, 2.3298]])
>>> torch.diag(a, 1)
tensor([[ 0.0000, 0.5950, 0.0000, 0.0000],
        [ 0.0000, 0.0000,-0.0872, 0.0000],
        [ 0.0000, 0.0000, 0.0000, 2.3298],
        [ 0.0000, 0.0000, 0.0000, 0.0000]])

获取给定矩阵的第 k 个对角线

>>> a = torch.randn(3, 3)
>>> a
tensor([[-0.4264, 0.0255,-0.1064],
        [ 0.8795,-0.2429, 0.1374],
        [ 0.1029,-0.6482,-1.6300]])
>>> torch.diag(a, 0)
tensor([-0.4264,-0.2429,-1.6300])
>>> torch.diag(a, 1)
tensor([ 0.0255, 0.1374])

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源