（beta）torch_npu.npu_one_hot

接口原型

torch_npu.npu_one_hot(input, num_classes=-1, depth=1, on_value=1, off_value=0) -> Tensor

功能描述

返回一个one-hot张量。input中index表示的位置采用on_value值，而其他所有位置采用off_value的值。

参数说明

• input (Tensor) - 任何shape的class值。
• num_classes (Int，默认值为-1) - 待填充的轴。
• depth (Int，默认值为1) - one_hot维度的深度。
• on_value (Scalar，默认值为1) - 当indices[j] == i时输出中的填充值。
• off_value (Scalar，默认值为0) - 当indices[j] != i时输出中的填充值。

调用示例

>>> a=torch.IntTensor([5, 3, 2, 1]).npu()
>>> b=torch_npu.npu_one_hot(a, depth=5)
>>> b
tensor([[0., 0., 0., 0., 0.],
        [0., 0., 0., 1., 0.],
        [0., 0., 1., 0., 0.],
        [0., 1., 0., 0., 0.]], device='npu:0')