Softmax2D¶

class paddle.nn. Softmax2D ( name=None ) ¶

Softmax2D 是 Softmax 的变体，其针对 3D 或者 4D 的 Tensor 在空间维度上计算 Softmax。具体来说，输出的 Tensor 的每个空间维度 \((channls, h_i, w_j)\) 求和为 1。

Softmax 的详细介绍请参考 Softmax