Extract a diagonal or construct a diagonal array.
View aliases
Main aliases
tf.keras.ops.diag(
x, k=0
)
Returns | |
---|---|
The extracted diagonal or constructed diagonal tensor. |
Examples:
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]])