one_hot

    该OP将输入’x’中的每个id转换为一个one-hot向量,其长度为 ,该id对应的向量维度上的值为1,其余维度的值为0。

    输出的Tensor的shape是在输入shape的最后一维后面添加了num_classes的维度。

    • 示例1:
    • x (Tensor) - 维度为

    • name (str,可选) – 具体用法请参见 ,一般无需设置,默认值为None。

    代码示例