mindspore.mint.nn.Dropout2d

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

During training, randomly zero some channels of the input tensor with probability p (For a 4-D tensor with a shape of \((N, C, H, W)\), the channel feature map refers to a 2-D feature map with the shape of \((H, W)\)).

For example, the \(j\) -th channel of the \(i\) -th sample in the batched input is a to-be-processed 2-D tensor input[i,j]. Each channel will be zeroed out independently on every forward call with probability p using samples from a Bernoulli distribution.

Dropout2d can improve the independence between channel feature maps.

Warning

This is an experimental API that is subject to change or deletion.

Refer to mindspore.mint.nn.functional.dropout2d() for more details.

Supported Platforms:

Ascend

Examples

>>> 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)