У меня есть входное изображение размером 128 x 96, и оно подается на net, после набора слоев я хочу вывести изображение того же размера.
Мой последний слой теперь - это ConvAct, и если проверить размер тензора, это (10, 1, 96, 128), поэтому иметь линейный слой, который вернет мне изображение 128 * 96 - какой размер ввода я должен использовать?
(12288, ...)?