tf.keras.backend.eye

TensorFlow 1 version View source on GitHub

Instantiate an identity matrix and returns it.

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:

kvar = tf.keras.backend.eye(3) 
tf.keras.backend.eval(kvar) 
array([[1.,  0.,  0.], 
       [0.,  1.,  0.], 
       [0.,  0.,  1.]], dtype=float32)