Вывод глубокой нейронной сети в Pytorch - PullRequest
Новая
       13

Вывод глубокой нейронной сети в Pytorch

1 голос
/ 03 августа 2020

У меня есть нейронная сеть, и это функция для нее: 

def mlp(sizes, activation=nn.Tanh, output_activation=nn.Identity):
    # Build a feedforward neural network. outputs are the logits
    layers = []
    for j in range(len(sizes)-1):
        act = activation if j < len(sizes)-2 else output_activation
        layers += [nn.Linear(sizes[j], sizes[j+1]), act()]
    return nn.Sequential(*layers)

и sizes=[10, 20]

Мои входные данные представляют собой тензор в форме [100,1,10] и с этим сеть, я получаю тензор в виде [100,20]. Я хотел бы добавить еще два слоя, чтобы вместо матрицы в качестве вывода я получил вектор размера 20. Возможно ли это, учитывая, что ввод включает в себя пакет изображений?

...