生成一个2D矩阵，主对角线是1，其他为0
【输入】
1个输入
x：2维tensor，用于拷贝tensor的shape
【输出】
一个输出
y：一个张量，和输入x同样的shape
【属性】
dtype：int，指定输出数据类型
k：int，默认是0，表示主对角线被广播成1的索引。如y是输出，则y[i, i+k] = 1
【约束】
仅支持k=0
Opset v8/v9/v10/v11/v12/v13/v14/v15/v16/v17/v18