tf.one_hot

Возвращает тензор one-hot.

tf.one_hot(
    indices,
    depth,
    on_value=None,
    off_value=None,
    axis=None,
    dtype=None,
    name=None
)

См. также tf.fill, tf.eye.

Позиции, представленные индексами в indices, принимают значение on_value, а все остальные позиции принимают значение off_value.

on_value и off_value должны иметь совпадающие типы данных. Если dtype также указан, они должны иметь тот же тип данных, что и dtype.

Если on_value не указан, он будет по умолчанию равен значению 1 типа dtype

Если off_value не указан, он будет по умолчанию равен значению 0 типа dtype

Если вход indices имеет ранг N, вывод будет иметь ранг N+1. Новая ось создается на позиции axis (по умолчанию новая ось добавляется в конец).

Если indices является скаляром, форма вывода будет вектором длины depth

Если indices является вектором длины features, форма вывода будет:

features x depth if axis == -1
depth x features if axis == 0

Если indices является матрицей (пакетной) с формой [batch, features], форма вывода будет:

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 является RaggedTensor, аргумент 'axis' должен быть положительным и ссылаться на неразрезанную ось. Вывод будет эквивалентен применению 'one_hot' к значениям RaggedTensor и созданию нового RaggedTensor из результата.

Если dtype не указан, он попытается принять тип данных on_value или off_value, если один или оба из них переданы. Если ни on_value, ни off_value, ни dtype не указаны, dtype по умолчанию будет равен tf.float32.

Примечание: Если требуется вывод с типом данных, отличным от числового (tf.string, tf.bool и т.д.), как on_value так и off_value обязательно должны быть переданы в one_hot.

Пример:

indices = [0, 1, 2]
depth = 3
tf.one_hot(indices, depth)  # output: [3 x 3]
# [[1., 0., 0.],
#  [0., 1., 0.],
#  [0., 0., 1.]]

indices = [0, 2, -1, 1]
depth = 3
tf.one_hot(indices, depth,
           on_value=5.0, off_value=0.0,
           axis=-1)  # output: [4 x 3]
# [[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
tf.one_hot(indices, depth,
           on_value=1.0, off_value=0.0,
           axis=-1)  # output: [2 x 2 x 3]
# [[[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)

indices = tf.ragged.constant([[0, 1], [2]])
depth = 3
tf.one_hot(indices, depth)  # output: [2 x None x 3]
# [[[1., 0., 0.],
#   [0., 1., 0.]],
#  [[0., 0., 1.]]]