tf.sparse.eye

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

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.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/sparse/eye