eye

    该OP用来构建二维Tensor，或一个批次的二维Tensor。

    参数：

    • batch_shape (list(int), 可选) - 如若提供，则返回Tensor的主批次维度将为batch_shape。

    • name (str) – 该参数供开发人员打印调试信息时使用，具体用法请参见 ，默认值为None。

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