mindspore.nn.Softmax2d

class mindspore.nn.Softmax2d[源代码]

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

将 Softmax 应用于具有shape \((C, H, W)\) 的输入Tensor的每个位置。

输入：

x (Tensor) - Tensor的shape \((N, C_{in}, H_{in}, W_{in})\) 或者 \((C_{in}, H_{in}, W_{in})\)。数据类型可以是float16或者float32。

输出：

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

异常：

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

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore
>>> from mindspore import Tensor, nn
>>> import numpy as np
>>> x = Tensor(np.array([[[[0.1, 0.2]], [[0.3, 0.4]], [[0.6, 0.5]]]]), mindspore.float32)
>>> softmax2d = nn.Softmax2d()
>>> output = softmax2d(x)
>>> print(output)
[[[[0.25838965 0.28001308]]
[[0.31559783 0.34200877]]
[[0.42601252 0.37797815]]]]