megengine.functional.one_hot

one_hot(inp, num_classes)

对输入张量进行 one-hot 编码。

参数

• inp (Tensor) – 输入张量。

• num_classes (int) – 表示输出张量最后一个维度的类数。

返回类型

Tensor

返回

输出张量。

例如：

import numpy as np
from megengine import tensor
import megengine.functional as F

x = tensor(np.arange(1, 4, dtype=np.int32))
out = F.one_hot(x, num_classes=4)
print(out.numpy())

输出：

[[0 1 0 0]
 [0 0 1 0]
 [0 0 0 1]]

megengine.functional.dropout megengine.functional.indexing_one_hot