torch.nn.functional.dropout2d

torch.nn.functional.dropout2d(input, p=0.5, training=True, inplace=False)

随机将整个通道（通道是二维特征图）置零。

例如，输入张量的批处理样本的第 $i$-th 个的第 $j$-th 通道是一个二维张量 $\text{input}[i, j]$。每个通道将在每次前向调用时，根据 Bernoulli 分布的采样，以 p 的概率独立地置零。

有关详细信息，请参阅 Dropout2d。

参数

• p (float) – 通道被置零的概率。默认为 0.5。

• training (bool) – 如果为 True，则应用 dropout。默认为 True

• inplace (bool) – 如果设置为 True，则执行此操作时将修改输入张量（原地操作）。默认为 False

返回类型

张量