tf.keras.ops.one_hot
Преобразует целочисленный тензор x в тензор с кодированием one-hot.
tf.keras.ops.one_hot(
x, num_classes, axis=-1, dtype=None, sparse=False
)
Кодирование one-hot — это представление, в котором каждое целочисленное значение преобразуется в двоичный вектор длиной num_classes, а индекс, соответствующий целочисленному значению, отмечается как 1, тогда как все остальные индексы отмечаются как 0.
| Аргументы | |
|---|---|
x | Целочисленный тензор для кодирования. Форма может быть произвольной, но тип данных должен быть целочисленным. |
num_classes | Количество классов для кодирования one-hot. |
axis | Ось, вдоль которой выполняется кодирование. По умолчанию используется -1, обозначающая последнюю ось. |
dtype | (Необязательно) Тип данных выходного тензора. Если не указан, используется тип данных backend по умолчанию. |
sparse | Возвращать ли разреженный тензор; применимо к backend, поддерживающим разреженные тензоры. |
| Возвращает | |
|---|---|
Целочисленный тензор: тензор с кодированием one-hot, имеющий ту же форму, что и x, за исключением указанного измерения axis, длина которого будет равна num_classes. Тип данных выходного тензора определяется значением dtype или типом данных backend по умолчанию. |
Пример:
x = keras.ops.convert_to_tensor([1, 3, 2, 0])
one_hot(x, num_classes=4)
array([[0. 1. 0. 0.]
[0. 0. 0. 1.]
[0. 0. 1. 0.]
[1. 0. 0. 0.]], shape=(4, 4), dtype=float32)
© 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/keras/ops/one_hot
