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] с единицами по диагонали. |
