tf.keras.ops.diag

Extract a diagonal or construct a diagonal array.

Main aliases

tf.keras.ops.numpy.diag

tf.keras.ops.diag(
    x, k=0
)

Args
`x`	Input tensor. If `x` is 2-D, returns the k-th diagonal of `x`. If `x` is 1-D, return a 2-D tensor with `x` on the k-th diagonal.
`k`	The diagonal to consider. Defaults to `0`. Use `k > 0` for diagonals above the main diagonal, and `k < 0` for diagonals below the main diagonal.

Returns
The extracted diagonal or constructed diagonal tensor.

from keras.src import ops
x = ops.arange(9).reshape((3, 3))
x
array([[0, 1, 2],
       [3, 4, 5],
       [6, 7, 8]])

ops.diag(x)
array([0, 4, 8])
ops.diag(x, k=1)
array([1, 5])
ops.diag(x, k=-1)
array([3, 7])

ops.diag(ops.diag(x)))
array([[0, 0, 0],
       [0, 4, 0],
       [0, 0, 8]])

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.

Last updated 2024-06-07 UTC.

Args