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.
|