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維進行

>>阅读原文<<