diag_embed

该 OP 创建一个 Tensor，其在指定的 2D 平面（由 和 指定）上的对角线由输入 填充。 默认的，指定的 2D 平面由返回 Tensor 的最后两维组成。

参数 确定在指定的二维平面中填充对角线的位置：

• 如果 offset = 0，则填充主对角线。

• 如果 offset < 0，则填充主对角线左下的对角线。

参数：

• dim2 （int ， 可选）- 填充对角线的二维平面的第二维，默认值为 -1。

返回： 指定二维平面填充了对角线的 Tensor。数据类型和输入数据类型一致。