diag

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

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

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

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

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

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

    1. import paddle
    3. x = paddle.to_tensor([[1, 2, 3], [4, 5, 6]])
    4. y = paddle.diag(x)
    5. # [1 5]
    7. y = paddle.diag(x, offset=1)
    8. print(y)
    9. # [2 6]
    11. y = paddle.diag(x, offset=-1)
    12. # [4]