torch.nn.functional.dropout2d

torch.nn.functional.dropout2d(input, p=0.5, training=True, inplace=False)[源代码]

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

例如，批处理输入中的第 $$i$$ 个样本的第 $$j$$ 个通道是一个二维张量 $\text{input}[i, j]$ ，它是输入张量的一部分。每个通道将在每次前向调用时以概率p独立地被置零，该概率是从伯努利分布中采样得到的。