在tensorflow中生成单位矩阵

tf.eye(num_rows, num_columns=None, dtype=tf.float32, name=None)

import tensorflow as tf

# 生成一个3x3的单位矩阵
identity_matrix = tf.eye(3, dtype=tf.float32)

with tf.Session() as sess:
    result = sess.run(identity_matrix)
    print(result)

[[1. 0. 0.]
 [0. 1. 0.]
 [0. 0. 1.]]