[Pytorch函数]torch.nn.dropout和torch.nn.dropout2d

# -*- coding: utf-8 -*-
import torch
import torch.nn as nn
import torch.autograd as autograd

m = nn.Dropout(p=0.5)
n = nn.Dropout2d(p=0.5)
input = autograd.Variable(torch.randn(2, 6, 3)) ## 对dim=1维进行随机置为0

print(m(input))
print('****************************************************')
print(n(input))

结果是：

可以看到torch.nn.Dropout对所有元素中每个元素按照概率0.5更改为零， 绿色椭圆，
而torch.nn.Dropout2d是对每个通道按照概率0.5置为0， 红色方框内
注：我只是圈除了部分