mindspore.nn.Softmax2d

class mindspore.nn.Softmax2d

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

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

输入：

• x (Tensor) - Tensor的shape \((N, C_{in}, H_{in}, W_{in})\) 或者 \((C_{in}, H_{in}, W_{in})\)。

输出：

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

异常：

• TypeError - x 的数据类型既不是float16也不是float32。

• ValueError - 数据格式不是“NCHW”或者“CHW”。

支持平台：

Ascend GPU CPU

样例：

>>> 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.258, 0.28]], [[0.316, 0.342]], [[0.426, 0.378]]]