tf.keras.ops.diag

Извлечь диагональ или создать диагональную матрицу.

Основные псевдонимы

tf.keras.ops.numpy.diag

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

Аргументы
`x`	Входной тензор. Если `x` является 2-мерным, возвращает k-ю диагональ `x`. Если `x` является 1-мерным, возвращает 2-мерный тензор с `x` на k-й диагонали.
`k`	Диагональ для рассмотрения. По умолчанию `0`. Используйте `k > 0` для диагоналей выше главной диагонали и `k < 0` для диагоналей ниже главной диагонали.

Возвращаемое значение
Извлеченная диагональ или построенный диагональный тензор.

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]])

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/keras/ops/diag