torch.diagonal¶
- torch.diagonal(input, offset=0, dim1=0, dim2=1) Tensor ¶
返回
input
的一个部分视图,其中包含其相对于dim1
和dim2
的对角线元素,作为形状末尾的一个维度附加。参数
offset
控制要考虑的对角线。如果
offset
= 0,则为主要对角线。如果
offset
> 0,则位于主对角线上方。如果
offset
< 0,则位于主对角线下方。
将
torch.diag_embed()
应用于此函数的输出,使用相同的参数会生成一个对角矩阵,其中包含输入的对角线条目。但是,torch.diag_embed()
具有不同的默认维度,因此需要显式指定这些维度。- 参数
注意
要获取批处理对角线,请传入 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]]])