tf.sparse.eye

Создаёт двумерный разреженный тензор с единицами по диагонали.

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

См. Руководство по миграции для получения дополнительной информации.

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

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

Возвращаемые значения
Разреженный тензор формы [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/versions/r2.9/api_docs/python/tf/sparse/eye