batch, 256, 256, 1 в первый слой. Второй вот что-то не хочет забирать выход первого, то есть 256, 256, 64 - наверное, там тоже нулевая размерность должна быть как размер батча, а вместо этого почему-то None Не понятно, чем первый код принципиально отличается от второго
Обсуждают сегодня