Как решать такие задачи?

Question

[minamisa]

Как решать такие задачи? Можете, пожалуйста, объяснить, а то не могу понять как это работает
Даны линейные слои:

слой_1 = nn.Linear(1024, 512)
слой_2 = nn.Linear(512, 128)
слой_3 = nn.Linear(128, 1024)
слой_4 = nn.Linear(???, ???)
Эти линейные слои соединены в нейронную сеть:
inp => слой_4 => слой_3 => слой_4 => out
Укажите правильный размер входного и выходного тензора у слой_4.

Это задание по PyTorch, Python

Answer 1

[Vindicar]

Размер выхода слоя равен размеру входа следующего слоя.
Вот как узнать размер выхода последнего слоя - я фз, тут недостаточно данных.

EDIT: ой, блин, я невнимательный. Не увидел, что слои идут не 1->2->3->4, а 4->3->4.
Тогда ещё проще, выход 4 должен совпадать по размеру со входом 3, а вход 4 - с выходом 3.

Comments

[U235U235]

все так, но т.к. слой 3 находится между двумя слоями 4, то их размерность будет очевидно (1024,128)

[Vindicar]

U235U235, я про выход слоя 4. ОП не привёл никаких подсказок, как должен выглядить слой out, так что нельзя определить, сколько выходов должно быть в слое 4 - только число входов.

EDIT: ой, блин, я невнимательный. Не увидел, что слои идут не 1->2->3->4, а 4->3->4.