2D卷积

代码

import torchimport torch.nn as nninput = torch.randn(1, 3, 112, 112)conv2d = nn.Conv2d(3, 64, (3, 3), 2, 1)output = conv2d(input)print(output.size())torch.Size([1, 64, 56, 56])

援用

<1>

torch=1.7.1+cu101torchvision=0.8.2torchaudio=0.7.2

<2>

https://pytorch.org/docs/1.7....