Join TensorFlow at Google I/O, May 11-12 Register now


TensorFlow 2 version View source on GitHub

Instantiate an identity matrix and returns it.

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

A Keras variable, an identity matrix.


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