tf.keras.ops.multi_hot
Кодирует целочисленные метки в векторы multi-hot.
tf.keras.ops.multi_hot(
inputs, num_classes=None, axis=-1, dtype=None, sparse=False, **kwargs
)
Эта функция кодирует целочисленные метки в векторы multi-hot, в которых каждой метке соответствует двоичное значение в результирующем векторе.
| Аргументы | |
|---|---|
inputs | Тензор целочисленных меток, которые нужно преобразовать в векторы multi-hot. |
num_classes | Целое число — общее количество уникальных классов. |
axis | (необязательно) Ось, вдоль которой следует добавить кодирование multi-hot. По умолчанию -1, что соответствует последнему измерению. |
dtype | (необязательно) Тип данных результирующего тензора. По умолчанию используется тип с плавающей точкой бэкенда. |
sparse | Следует ли возвращать разреженный тензор; для бэкендов, поддерживающих разреженные тензоры. |
| Возвращает | |
|---|---|
Tensor | Тензор, закодированный в формате multi-hot. |
Пример:
data = keras.ops.convert_to_tensor([0, 4]) keras.ops.multi_hot(data, num_classes=5) array([1.0, 0.0, 0.0, 0.0, 1.0], 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/multi_hot
