tf.eye

Создает единичную матрицу или пакет матриц.

Просмотреть псевдонимы

Основные псевдонимы

tf.linalg.eye

Псевдонимы для обеспечения совместимости при миграции

Подробнее см. в руководстве по миграции.

tf.compat.v1.eye, tf.compat.v1.linalg.eye

tf.eye(
    num_rows,
    num_columns=None,
    batch_shape=None,
    dtype=tf.dtypes.float32,
    name=None
)

Используется в блокнотах

Используется в руководстве	Используется в учебных материалах
Расширенное автоматическое дифференцирование Введение в графы и tf.function	Практический пример TensorFlow Probability: оценка ковариации Зоопарк обучаемых распределений Обзор TensorFlow Probability Байесовская гауссовская модель смеси и гамильтоновский MCMC Байесовское моделирование с помощью Joint Distribution

См. также tf.ones, tf.zeros, tf.fill, tf.one_hot.

# Construct one identity matrix.
tf.eye(2)
==> [[1., 0.],
     [0., 1.]]

# Construct a batch of 3 identity matrices, each 2 x 2.
# batch_identity[i, :, :] is a 2 x 2 identity matrix, i = 0, 1, 2.
batch_identity = tf.eye(2, batch_shape=[3])

# Construct one 2 x 3 "identity" matrix
tf.eye(2, num_columns=3)
==> [[ 1.,  0.,  0.],
     [ 0.,  1.,  0.]]

Аргументы
`num_rows`	Неотрицательный скаляр `int32` типа `Tensor`, задающий число строк в каждой матрице пакета.
`num_columns`	Необязательный неотрицательный скаляр `int32` типа `Tensor`, задающий число столбцов в каждой матрице пакета. По умолчанию — `num_rows`.
`batch_shape`	Список или кортеж целых чисел Python либо одномерный `int32` `Tensor`. Если указан, возвращаемый `Tensor` будет иметь начальные пакетные размерности этой формы.
`dtype`	Тип элемента в результирующем `Tensor`
`name`	Имя этого `Op`. По умолчанию — "eye".

Возвращает
`Tensor` формы `batch_shape + [num_rows, num_columns]`

© 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/eye