mindspore.ops.one_hot
=====================

.. image:: https://mindspore-website.obs.cn-north-4.myhuaweicloud.com/website-images/br_base/resource/_static/logo_source.svg
    :target: https://gitee.com/mindspore/mindspore/blob/br_base/docs/api/api_python/ops/mindspore.ops.func_one_hot.rst
    :alt: 查看源文件


.. py:function:: mindspore.ops.one_hot(indices, depth, on_value=1, off_value=0, axis=-1)

    生成一个新的tensor，索引 `indices` 表示的位置上取值 `on_value` ，其他所有位置上取值 `off_value` 。

    .. note::
        如果 `indices` 的秩为 `n` ，则输出Tensor的秩为 `n+1` 。新轴在 `axis` 处创建。当执行设备是 Ascend 时，如果 `on_value` 为int64类型，则 `indices` 也必须为int64类型，且 `on_value` 和 `off_value` 的取值只能是1和0。

    参数：
        - **indices** (Tensor) - 输入索引。
        - **depth** (int) - one-hot深度。
        - **on_value** (Union[Tensor, int, float]，可选) - 填充索引位置上的值。默认 ``1``。
        - **off_value** (Union[Tensor, int, float]，可选) - 填充非索引位置上的值。默认 ``0``。
        - **axis** (int，可选) - 指定计算轴。默认 ``-1`` 。

    返回：
        Tensor