tf.keras.ops.diagonal
Возвращает указанные диагонали.
tf.keras.ops.diagonal(
x, offset=0, axis1=0, axis2=1
)
Если x является 2-мерным, возвращает диагональ x с заданным смещением, т.е. коллекцию элементов вида x[i, i+offset].
Если у x более двух измерений, оси, указанные axis1 и axis2, используются для определения 2-мерного подмассива, диагональ которого возвращается.
Форма результирующего массива может быть определена путем удаления axis1 и axis2 и добавления индекса справа, равного размеру результирующих диагоналей.
| Аргументы | |
|---|---|
x | Входной тензор. |
offset | Смещение диагонали от главной диагонали. Может быть положительным или отрицательным. По умолчанию равно 0 (главная диагональ). |
axis1 | Ось, используемая в качестве первой оси 2-мерных подмассивов. По умолчанию равна 0 (первая ось). |
axis2 | Ось, используемая в качестве второй оси 2-мерных подмассивов. По умолчанию равна 1 (вторая ось). |
| Возвращаемое значение | |
|---|---|
| Тензор диагоналей. |
Примеры:
from keras.src import ops
x = ops.arange(4).reshape((2, 2))
x
array([[0, 1],
[2, 3]])
x.diagonal()
array([0, 3])
x.diagonal(1)
array([1])x = ops.arange(8).reshape((2, 2, 2))
x
array([[[0, 1],
[2, 3]],
[[4, 5],
[6, 7]]])
x.diagonal(0, 0, 1)
array([[0, 6],
[1, 7]])
© 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/diagonal
