Missed TensorFlow World? Check out the recap. Learn more

tf.sparse.eye

TensorFlow 2.0 version View source on GitHub

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

Aliases:

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