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维进行

>>阅读原文<<