torch.diagflat¶
- torch.diagflat(input, offset=0) Tensor ¶
如果
input
是向量(一维张量),则返回一个二维方阵张量,其元素为input
的对角线元素。如果
input
是一个具有多个维度的张量,则返回一个二维张量,其对角线元素等于input
的扁平化结果。
参数
offset
控制要考虑的对角线。如果
offset
= 0,则为主对角线。如果
offset
> 0,则为主对角线上方。如果
offset
< 0,则为主对角线下方。
示例
>>> a = torch.randn(3) >>> a tensor([-0.2956, -0.9068, 0.1695]) >>> torch.diagflat(a) tensor([[-0.2956, 0.0000, 0.0000], [ 0.0000, -0.9068, 0.0000], [ 0.0000, 0.0000, 0.1695]]) >>> torch.diagflat(a, 1) tensor([[ 0.0000, -0.2956, 0.0000, 0.0000], [ 0.0000, 0.0000, -0.9068, 0.0000], [ 0.0000, 0.0000, 0.0000, 0.1695], [ 0.0000, 0.0000, 0.0000, 0.0000]]) >>> a = torch.randn(2, 2) >>> a tensor([[ 0.2094, -0.3018], [-0.1516, 1.9342]]) >>> torch.diagflat(a) tensor([[ 0.2094, 0.0000, 0.0000, 0.0000], [ 0.0000, -0.3018, 0.0000, 0.0000], [ 0.0000, 0.0000, -0.1516, 0.0000], [ 0.0000, 0.0000, 0.0000, 1.9342]])