tf.keras.backend.eye

View source on GitHub

Instantiate an identity matrix and returns it.

View aliases

Compat aliases for migration

See Migration guide for more details.

tf.compat.v1.keras.backend.eye, `tf.compat.v2.keras.backend.eye`

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

Arguments
size	Integer, number of rows/columns.
dtype	String, data type of returned Keras variable.
name	String, name of returned Keras variable.

Returns
A Keras variable, an identity matrix.

Example:

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)