Функция eye
Функция eye создает единичную матрицу заданного размера.
Первым параметром функция принимает количество строк,
вторым - количество столбцов (если не указано,
создается квадратная матрица). Третьим параметром
можно передать тип данных элементов.
Также доступен параметр name для именования операции.
Синтаксис
tf.eye(num_rows, [num_columns], [dtype], [name])
Пример
Давайте создадим единичную матрицу размером 3 на 3:
import tensorflow as tf
t = tf.eye(3)
print(t)
Результат выполнения кода:
tf.Tensor(
[[1. 0. 0.]
[0. 1. 0.]
[0. 0. 1.]], shape=(3, 3), dtype=float32)
Пример
Давайте создадим прямоугольную единичную матрицу размером 2 на 4:
import tensorflow as tf
t = tf.eye(2, 4)
print(t)
Результат выполнения кода:
tf.Tensor(
[[1. 0. 0. 0.]
[0. 1. 0. 0.]], shape=(2, 4), dtype=float32)
Пример
Давайте создадим единичную матрицу с целочисленным типом int32:
import tensorflow as tf
t = tf.eye(3, dtype=tf.int32)
print(t)
Результат выполнения кода:
tf.Tensor(
[[1 0 0]
[0 1 0]
[0 0 1]], shape=(3, 3), dtype=int32)
