tf.linalg.tensor_diag  |  TensorFlow v2.16.1 (original) (raw)

tf.linalg.tensor_diag

Stay organized with collections Save and categorize content based on your preferences.

Returns a diagonal tensor with a given diagonal values.

View aliases

Compat aliases for migration

SeeMigration guide for more details.

tf.compat.v1.diag, tf.compat.v1.linalg.tensor_diag

tf.linalg.tensor_diag(
    diagonal: Annotated[Any, TV_Diag_T], name=None
) -> Annotated[Any, TV_Diag_T]

Given a diagonal, this operation returns a tensor with the diagonal and everything else padded with zeros. The diagonal is computed as follows:

Assume diagonal has dimensions [D1,..., Dk], then the output is a tensor of rank 2k with dimensions [D1,..., Dk, D1,..., Dk] where:

output[i1,..., ik, i1,..., ik] = diagonal[i1, ..., ik] and 0 everywhere else.

For example:

# 'diagonal' is [1, 2, 3, 4]
tf.diag(diagonal) ==> [[1, 0, 0, 0]
                       [0, 2, 0, 0]
                       [0, 0, 3, 0]
                       [0, 0, 0, 4]]