TensorFlow函数教程：tf.keras.backend.eye

tf.keras.backend.eye函数

tf.keras.backend.eye(     size,     dtype=None,     name=None ) 

定义在：tensorflow/python/keras/backend.py。

实例化一个单位矩阵并返回它。

参数：

• size：整数，行数/列数。
• dtype：字符串，返回的Keras变量的数据类型。
• name：字符串，返回的Keras变量的名称。

返回：

返回一个Keras变量，一个单位矩阵。

示例： 

>>> from keras import backend as K >>> kvar = K.eye(3) >>> K.eval(kvar) array([[ 1., 0., 0.], [ 0., 1., 0.], [ 0., 0., 1.]], dtype=float32)