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
