tf.keras.ops.trace
Возвращает сумму элементов тензора вдоль диагоналей.
tf.keras.ops.trace(
x, offset=0, axis1=0, axis2=1
)
Если x — двумерный массив, возвращается сумма элементов его диагонали с указанным смещением, то есть сумма элементов x[i, i+offset] для всех i.
Если массив a имеет более двух измерений, то оси, заданные параметрами axis1 и axis2, используются для определения двумерных подмассивов, следы которых возвращаются.
Форма результирующего тензора совпадает с формой x, из которой удалены axis1 и axis2.
| Аргументы | |
|---|---|
x | Входной тензор. |
offset | Смещение диагонали относительно главной диагонали. Может быть положительным или отрицательным. По умолчанию — 0. |
axis1 | Ось, используемая в качестве первой оси двумерных подмассивов. По умолчанию — 0.(первая ось). |
axis2 | Ось, используемая в качестве второй оси двумерных подмассивов. По умолчанию — 1 (вторая ось). |
| Возвращает | |
|---|---|
Если x — двумерный массив, возвращается сумма элементов диагонали. Если x имеет больше измерений, возвращается тензор с суммами элементов вдоль диагоналей. |
© 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/trace
