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