tf.keras.ops.tri
Возвращает тензор с единицами на диагонали и ниже неё и нулями в остальных позициях.
tf.keras.ops.tri(
N, M=None, k=0, dtype=None
)
| Аргументы | |
|---|---|
N | Количество строк в тензоре. |
M | Количество столбцов в тензоре. |
k | Поддиагональ, на которой и ниже которой заполняется массив. k = 0 — это главная диагональ, k < 0 находится ниже неё, а k > 0 — выше. Значение по умолчанию — 0. |
dtype | Тип данных возвращаемого тензора. Значение по умолчанию — "float32". |
| Возвращает | |
|---|---|
Тензор, в котором нижний треугольник заполнен единицами, а остальные элементы — нулями. T[i, j] == 1 для j <= i + k, в противном случае — 0. |
© 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/tri
