tf.sparse.eye
Creates a two-dimensional sparse tensor with ones along the diagonal.
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. |