PyTorch: получить размер входного слоя

Question

Я хочу программно найти размер моего входного слоя.

Если мой первый слой называется fc1, как мне узнать его входные данные?

Answer 1

Если ваша модель называется model, это даст количество входных объектов слоя fc1:

model.fc1.in_features

Это полезно внутри метода .forward():

def forward(self, x):
    x = x.view(-1, self.fc1.in_features)  # resize the input to match the input layer
    ...