torch.nn.functional.one_hot

torch.nn.functional.one_hot(tensor, num_classes=- 1) → LongTensor

Принимает LongTensor со значениями индексов формы (*) и возвращает тензор формы (*, num_classes), в котором везде стоят нули, кроме места, где индекс последнего измерения соответствует соответствующему значению входного тензора, в этом случае он будет 1.

См. также One-hot на Википедии .

Параметры:

• tensor (LongTensor) – значения классов любой формы.
• num_classes (int) – Общее количество классов. Если установлено в -1, количество классов будет определено как на единицу больше, чем наибольшее значение класса во входном тензоре.

Возвращает:

LongTensor, имеющий на один размер больше с 1 значениями в индексе последнего измерения, указанного вводом, и 0 везде в другом месте.

Примеры

>>> F.one_hot(torch.arange(0, 5) % 3)
tensor([[1, 0, 0],
        [0, 1, 0],
        [0, 0, 1],
        [1, 0, 0],
        [0, 1, 0]])
>>> F.one_hot(torch.arange(0, 5) % 3, num_classes=5)
tensor([[1, 0, 0, 0, 0],
        [0, 1, 0, 0, 0],
        [0, 0, 1, 0, 0],
        [1, 0, 0, 0, 0],
        [0, 1, 0, 0, 0]])
>>> F.one_hot(torch.arange(0, 6).view(3,2) % 3)
tensor([[[1, 0, 0],
         [0, 1, 0]],
        [[0, 0, 1],
         [1, 0, 0]],
        [[0, 1, 0],
         [0, 0, 1]]])