TensorFlow函數(shù)教程：tf.keras.backend.eye

tf.keras.backend.eye函數(shù)

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

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

實(shí)例化一個(gè)單位矩陣并返回它。

參數(shù)：

• size：整數(shù)，行數(shù)/列數(shù)。
• dtype：字符串，返回的Keras變量的數(shù)據(jù)類型。
• name：字符串，返回的Keras變量的名稱。

返回：

返回一個(gè)Keras變量，一個(gè)單位矩陣。

示例： 

>>> 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)