tf.keras.ops.triu
Возвращает верхний треугольник тензора.
tf.keras.ops.triu(
x, k=0
)
Для тензоров с ndim больше 2 операция triu применяется к двум последним осям.
| Аргументы | |
|---|---|
x | Входной тензор. |
k | Диагональ, ниже которой элементы обнуляются. По умолчанию 0 — главная диагональ. k < 0 находится ниже неё, а k > 0 — выше. |
| Возвращает | |
|---|---|
Верхний треугольник 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/triu