PyTorch 深度學習:30分鐘快速入門

卷積¶ 卷積在 pytorch 中有兩種方式,一種是 torch.nn.Conv2d(),一種是 torch.nn.functional.conv2d(),這兩種形式本質都是使用一個卷積操作 這兩種形式的卷積對於輸入的要求都是一樣的,首先需要輸入是一個 torch.autograd.Variable() 的類型,大小是 (batch, channel, H, W),其中 batch 表示輸入的一批
相關文章
相關標籤/搜索