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)