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})\)。 数据类型可以是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]]]]