В общем, в модельке, которая будет ML-моделькой, нужно в конструкторе сначала создавать функции, КОТОРЫМИ потом вы будете создавать тензоры. Если вы будете создавать тензоры напрямую функциями высшего порядка в call, эти тензоры не попадут ни в поле "слои" модельики ни в "мутируемые переменные" (веса сетей), и по ним не будет производиться обучение. Какая же кривота... При этом на функцию-создание тензора может приходиться только один шейп входящего тензора. Если одну и ту же функцию создания тензора вызвать на двух разных по форме тензорах, будет падение кода. Хочешь применять одну и ту же функцию на разных слоях — создавай несколько таких
