mindspore.mint.nn.Dropout2d

class mindspore.mint.nn.Dropout2d(p=0.5, inplace=False)

在训练期间，以概率 p 随机将输入 tensor 的某些通道置零（对于shape为 \((N, C, H, W)\) 的四维 tensor，其通道特征图指的是后两维shape为 \((H, W)\) 的二维特征图）。

例如，在批处理输入中，第 \(i\) 批的第 \(j\) 个通道是一个待处理的二维 tensor input[i, j]。 每个通道在每次前向传播时，将会独立地根据伯努利分布的概率 p 来确定是否被置零。

Dropout2d 可以提高通道特征映射之间的独立性。

警告

这是一个实验性API，后续可能修改或删除。

更多参考详见 mindspore.mint.nn.functional.dropout2d()。

支持平台：

Ascend

样例：

>>> import mindspore
>>> dropout = mindspore.mint.nn.Dropout2d(p=0.5)
>>> x = mindspore.mint.ones(([2, 1, 2, 3]), dtype=mindspore.float32)
>>> output = dropout(x)
>>> print(output.shape)
(2, 1, 2, 3)