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

Question

Мне было интересно, есть ли много сверточных слоев (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?

Answer 1

Вторым параметром конструктора nn.Conv2D является количество выходных каналов:

self.conv1 = nn.Conv2D(in_channels,conv1_out_channels)
self.conv2 = nn.Conv2D(conv1_out_channels,out_ch2)

, как описано в документах

Также доступно как свойство:

self.conv1.out_channels