mindspore.nn.Softmax2d
======================

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


.. py:class:: mindspore.nn.Softmax2d()

    应用于2D特征数据的Softmax函数。

    将 `Softmax` 应用于具有shape :math:`(C, H, W)` 的输入Tensor的每个位置 :math:`(c, h, w)` 。

    输入：
        - **x** (Tensor) - Tensor的shape :math:`(N, C_{in}, H_{in}, W_{in})` 或者 :math:`(C_{in}, H_{in}, W_{in})`。
          数据类型可以是float16或者float32。

    输出：
        Tensor，数据类型和shape与 `x` 相同，取值范围为[0, 1]。

    异常：
        - **TypeError** - `x` 的数据类型既不是float16也不是float32。
        - **ValueError** - 数据格式不是“NCHW”或者“CHW”。