快捷方式

torch.diagonal

torch.diagonal(input, offset=0, dim1=0, dim2=1) Tensor

返回 input 的一个部分视图,其中包含其相对于 dim1dim2 的对角线元素,作为形状末尾的一个维度附加。

参数 offset 控制要考虑的对角线。

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

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

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

torch.diag_embed() 应用于此函数的输出,使用相同的参数会生成一个对角矩阵,其中包含输入的对角线条目。但是,torch.diag_embed() 具有不同的默认维度,因此需要显式指定这些维度。

参数
  • input (Tensor) – 输入张量。必须至少是二维的。

  • offset (int, 可选) – 要考虑的对角线。默认值:0(主对角线)。

  • dim1 (int, 可选) – 用于获取对角线的第一个维度。默认值:0。

  • dim2 (int, 可选) – 用于获取对角线的第二个维度。默认值:1。

注意

要获取批处理对角线,请传入 dim1=-2,dim2=-1。

示例

>>> a = torch.randn(3, 3)
>>> a
tensor([[-1.0854,  1.1431, -0.1752],
        [ 0.8536, -0.0905,  0.0360],
        [ 0.6927, -0.3735, -0.4945]])


>>> torch.diagonal(a, 0)
tensor([-1.0854, -0.0905, -0.4945])


>>> torch.diagonal(a, 1)
tensor([ 1.1431,  0.0360])


>>> x = torch.randn(2, 5, 4, 2)
>>> torch.diagonal(x, offset=-1, dim1=1, dim2=2)
tensor([[[-1.2631,  0.3755, -1.5977, -1.8172],
         [-1.1065,  1.0401, -0.2235, -0.7938]],

        [[-1.7325, -0.3081,  0.6166,  0.2335],
         [ 1.0500,  0.7336, -0.3836, -1.1015]]])

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发者的深入教程

查看教程

资源

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

查看资源