diag

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

    参数：

    • 一维Tensor，会把该Tensor的元素赋在方阵的对角线上。数据类型可以是 float32，float64，int32，int64。

    返回类型：Variable，数据类型和输入数据类型一致。

    代码示例：