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