PaddlePaddle

 paddle / diag


diag

paddle. diag ( x, offset=0, padding_value=0, name=None ) [源代码]

如果 x 是向量(1-D张量),则返回带有 x 元素作为对角线的2-D方阵。

如果 x 是矩阵(2-D张量),则提取 x 的对角线元素,以1-D张量返回。

参数 offset 控制对角线偏移量:

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

  • 如果 offset > 0,则为上对角线。

  • 如果 offset < 0,则为下对角线。

参数

  • x(Tensor):输入的 Tensor。它的形状可以是一维或二维。其数据类型应为float32,float64,int32,int64。

  • offset(int,可选):对角线偏移量。正值表示上对角线,0表示主对角线,负值表示下对角线。

  • padding_value(int|float,可选):使用此值来填充指定对角线以外的区域。仅在输入为一维张量时生效。默认值为0。

  • name (str,可选) - 操作的名称(可选,默认值为None)。更多信息请参见 Name

返回

Tensor,方阵或向量。输出数据类型与输入数据类型相同。

代码示例 1

import paddle

x = paddle.to_tensor([1, 2, 3])
y = paddle.diag(x)
print(y)
# [[1 0 0]
#  [0 2 0]
#  [0 0 3]]

y = paddle.diag(x, offset=1)
print(y)
# [[0 1 0 0]
#  [0 0 2 0]
#  [0 0 0 3]
#  [0 0 0 0]]

y = paddle.diag(x, padding_value=6)
print(y)
# [[1 6 6]
#  [6 2 6]
#  [6 6 3]]

代码示例 2

import paddle

x = paddle.to_tensor([[1, 2, 3], [4, 5, 6]])
y = paddle.diag(x)
print(y)
# [1 5]

y = paddle.diag(x, offset=1)
print(y)
# [2 6]

y = paddle.diag(x, offset=-1)
print(y)
# [4]

此页内容是否对您有帮助