tf.keras.ops.eye
Возвращает 2-мерный тензор с единицами на диагонали и нулями в других местах.
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
