Функция identity
Функция identity создает квадратную единичную матрицу.
Первым параметром функция принимает размер матрицы.
Вторым параметром можно передать тип данных элементов.
Единичная матрица содержит единицы на главной диагонали,
а все остальные элементы равны нулю.
Синтаксис
tf.identity(num, [dtype])
Пример
Давайте создадим единичную матрицу размером 3 на 3:
import tensorflow as tf
t = tf.identity(3)
print(t)
Результат выполнения кода:
tf.Tensor(
[[1. 0. 0.]
[0. 1. 0.]
[0. 0. 1.]], shape=(3, 3), dtype=float32)
Пример
Давайте создадим единичную матрицу с типом int32:
import tensorflow as tf
t = tf.identity(4, dtype=tf.int32)
print(t)
Результат выполнения кода:
tf.Tensor(
[[1 0 0 0]
[0 1 0 0]
[0 0 1 0]
[0 0 0 1]], shape=(4, 4), dtype=int32)
Пример
Давайте проверим размер и тип созданной матрицы:
import tensorflow as tf
t = tf.identity(5)
print(t.shape)
print(t.dtype)
Результат выполнения кода:
(5, 5)
<dtype: 'float32'>
