PyTorch_Conv2d

2D卷积 PyTorch_Conv2d
文章图片

代码

import torch import 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+cu101 torchvision=0.8.2 torchaudio=0.7.2

<2>
【PyTorch_Conv2d】https://pytorch.org/docs/1.7....

    推荐阅读