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 | (Необязательно) Тип данных выходного тензора. Если не указан, он по умолчанию соответствует типу данных бэкенда. |
sparse | Нужно ли возвращать разреженный тензор; для бэкэндов, которые поддерживают разреженные тензоры. |
| Возвращаемое значение | |
|---|---|
Целочисленный тензор: Тензор с one-hot кодированием, имеющий такую же форму, как и x, за исключением указанной axis размерности, которая будет иметь длину num_classes. Тип данных выходного тензора определяется значением dtype или типом данных бэкенда по умолчанию. |
Пример:
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
