快捷方式

Softmax2d

class torch.nn.Softmax2d(*args, **kwargs)[source]

对每个空间位置的特征应用 SoftMax。

给定一个 Channels x Height x Width 的图像,它将对每个位置 (Channels,hi,wj)(Channels, h_i, w_j) 应用 Softmax

形状
  • 输入:(N,C,H,W)(N, C, H, W)(C,H,W)(C, H, W).

  • 输出:(N,C,H,W)(N, C, H, W)(C,H,W)(C, H, W)(与输入形状相同)

返回值

一个与输入维度和形状相同的张量,其值在 [0, 1] 范围内。

返回类型

None

示例

>>> m = nn.Softmax2d()
>>> # you softmax over the 2nd dimension
>>> input = torch.randn(2, 3, 12, 13)
>>> output = m(input)

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发者的深入教程

查看教程

资源

查找开发资源并获得解答

查看资源