2D 卷积
代码
import torch
import torch.nn as nn
input = 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+cu101
torchvision=0.8.2
torchaudio=0.7.2<2>
https://pytorch.org/docs/1.7….