tf.sparse.eye

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

Compat aliases for migration

See Migration guide for more details.

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.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.

Last updated 2021-08-16 UTC.

Args