tf.sparse.eye

Creates a two-dimensional sparse tensor with ones along the diagonal.

View aliases

Compat aliases for migration

See Migration guide for more details.

tf.compat.v1.sparse.eye, `tf.compat.v2.sparse.eye`

tf.sparse.eye(
    num_rows, num_columns=None, dtype=tf.dtypes.float32, name=None
)

Args
num_rows	Non-negative integer or int32 scalar tensor giving the number of rows in the resulting matrix.
num_columns	Optional non-negative integer or int32 scalar tensor giving the number of columns in the resulting matrix. Defaults to num_rows.
dtype	The type of element in the resulting Tensor.
name	A name for this Op. Defaults to "eye".

Returns
A SparseTensor of shape [num_rows, num_columns] with ones along the diagonal.