EyeLike

功能

生成一个2D矩阵，主对角线是1，其他为0。

输入

一个输入：

x：2维tensor，用于拷贝tensor的shape。

输出

一个输出：

y：一个tensor，和输入x同样的shape。

属性

dtype：int，指定输出数据类型，详情请参考https://onnx.ai/onnx/api/mapping.html。

k：int，默认是0，表示主对角线被广播成1的索引。如y是输出，则y[i, i+k] = 1。

约束

仅支持k=0。

支持的ONNX版本

Opset v8/v9/v10/v11/v12/v13/v14/v15/v16/v17/v18