Получить входные каналы для conv2d с предыдущего слоя?

Мне было интересно, много ли сверточных слоев (conv1 -> conv2). Как мы можем получить параметр входных каналов для conv2 из выходного канала conv1?

class MyModel(nn.Module): def __init__(self, in_ch, num_features, out_ch2): super(MyModel, self).__init__() self.conv1 = nn.Conv2D(in_channels,num_features) self.conv2 = nn.Conv2D(in_channnels_from_out_channels_of_conv1,out_ch2)

Могу ли я получить out_channels из слоя conv1 и использовать его как in_ch для conv2?

Источник: ledsshop.ru