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