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]]])