tf.keras.ops.eye
Возвращает двумерный тензор с единицами на диагонали и нулями в остальных элементах.
tf.keras.ops.eye(
N, M=None, k=0, dtype=None
)
| Аргументы | |
|---|---|
N | Количество строк в выходных данных. |
M | Количество столбцов в выходных данных. Если None, по умолчанию используется N. |
k | Индекс диагонали: 0 (значение по умолчанию) соответствует главной диагонали, положительное значение — верхней диагонали, а отрицательное — нижней диагонали. |
dtype | Тип данных возвращаемого тензора. |
| Возвращает | |
|---|---|
| Тензор с единицами на k-й диагонали и нулями в остальных элементах. |
© 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/eye
