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)
  >>> btensor([[0., 0., 0., 0., 0.],
          [0., 0., 0., 1., 0.],
          [0., 0., 1., 0., 0.],
          [0., 1., 0., 0., 0.]], device='npu:0')