Missed TensorFlow World? Check out the recap. Learn more

tf.keras.backend.eye

TensorFlow 2.0 version View source on GitHub

Instantiate an identity matrix and returns it.

Aliases:

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