eye

    该OP用来构建二维Tensor(主对角线元素为1，其他元素为0)。

    参数：

    • num_rows (int) - 生成2-D Tensor的行数，数据类型为非负int32。

    • dtype (np.dtype|str， 可选) - 返回Tensor的数据类型，可为float16，float32，float64， int32， int64。若为None, 则默认等于float32。

    返回： shape 为 [num_rows, num_columns]的Tensor。

    • : - 如果 dtype 的类型不是float16， float32， float64， int32， int64其中之一。

    代码示例：