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».

Возвращает
`SparseTensor` с формой [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/sparse/eye