PyTorch快速搭建多个具有相同卷积核尺寸不同卷积核数量的卷积层

Posted 2023-01-12 算法与编程之美

问题

本文主要介绍啊快速搭建多个具有相同卷积核尺寸不同卷积核数量的卷积层。

方法

快速搭建三个卷积层结构，每个卷积层的参数如下所示：

• 3@3x3
• 16@3x3
• 32@3x3
• 64@3x3

x@yxy 其中x为卷积核数量,y为卷积核大小

简单方法

import torch
from torch import nn

# (1) init
conv1 = nn.Conv2d(
    in_channels=3,
    out_channels=16,
    kernel_size=