diag

    该OP创建一个方阵,使用输入diagonal来指定方阵的对角线元素的值。

    存储着方阵的Tensor,对角线值是输入Tensor diagonal的值, 数据shape为

    二维Tensor。

    Variable,数据类型和输入数据类型一致。