Что будет с тензором размером 16 на 5 на 5, если применить к нему flatten? Мы проходили её на семинаре об FC-MNIST (шаг 2), только там это была операция reshape, что -- то же самое, что и flatten, но с возможностью указать, сколько будет слоёв в итоговом тензоре.