Возвращает горячий тензор.
Местоположение, представленное индексами в `indices`, принимает значение `on_value`, а все остальные местоположения принимают значение `off_value`.
Если входные «индексы» имеют ранг «N», выходные данные будут иметь ранг «N+1». Новая ось создается в измерении «ось» (по умолчанию: новая ось добавляется в конце).
Если индексы являются скалярами, выходная форма будет вектором длины «глубина».
Если «индексы» — это вектор длины «функции», выходная форма будет следующей:
features x depth if axis == -1
depth x features if axis == 0
batch x features x depth if axis == -1
batch x depth x features if axis == 1
depth x batch x features if axis == 0
Предположим, что
indices = [0, 2, -1, 1]
depth = 3
on_value = 5.0
off_value = 0.0
axis = -1
output =
[5.0 0.0 0.0] // one_hot(0)
[0.0 0.0 5.0] // one_hot(2)
[0.0 0.0 0.0] // one_hot(-1)
[0.0 5.0 0.0] // one_hot(1)
indices = [0, 2, -1, 1]
depth = 3
on_value = 0.0
off_value = 3.0
axis = 0
output =
[0.0 3.0 3.0 3.0]
[3.0 3.0 3.0 0.0]
[3.0 3.0 3.0 3.0]
[3.0 0.0 3.0 3.0]
// ^ one_hot(0)
// ^ one_hot(2)
// ^ one_hot(-1)
// ^ one_hot(1)
indices = [[0, 2], [1, -1]]
depth = 3
on_value = 1.0
off_value = 0.0
axis = -1
output =
[
[1.0, 0.0, 0.0] // one_hot(0)
[0.0, 0.0, 1.0] // one_hot(2)
][
[0.0, 1.0, 0.0] // one_hot(1)
[0.0, 0.0, 0.0] // one_hot(-1)
]
Вложенные классы
сорт | OneHot.Параметры | Дополнительные атрибуты для OneHot |
Константы
Нить | OP_NAME | Название этой операции, известное основному движку TensorFlow. |
Публичные методы
Выход <U> | какВывод () Возвращает символический дескриптор тензора. |
статический OneHot.Options | ось (длинная ось) |
статический <U расширяет TType > OneHot <U> | |
Выход <U> | выход () Одногорячий тензор. |
Унаследованные методы
Константы
общедоступная статическая финальная строка OP_NAME
Название этой операции, известное основному движку TensorFlow.
Публичные методы
публичный вывод <U> asOutput ()
Возвращает символический дескриптор тензора.
Входные данные для операций TensorFlow являются выходными данными другой операции TensorFlow. Этот метод используется для получения символического дескриптора, который представляет собой вычисление входных данных.
общедоступная статическая ось OneHot.Options (длинная ось)
Параметры
ось | Ось для заполнения (по умолчанию: -1, новая самая внутренняя ось). |
---|
public static OneHot <U> create (область области действия , операнд <? расширяет индексы TNumber >, операнд < TInt32 > глубина, операнд <U> onValue, операнд <U> offValue, параметры... параметры)
Фабричный метод для создания класса, обертывающего новую операцию OneHot.
Параметры
объем | текущий объем |
---|---|
индексы | Тензор индексов. |
глубина | Скаляр, определяющий глубину одного горячего измерения. |
onValue | Скаляр, определяющий значение для заполнения вывода, когда `indices[j] = i`. |
OffValue | Скаляр, определяющий значение для заполнения вывода, когда `indices[j] != i`. |
параметры | содержит значения необязательных атрибутов |
Возврат
- новый экземпляр OneHot